Skip to content

Commit a0090e8

Browse files
authored
Merge pull request #48 from scaleapi/sasha/upload_fix
change test files to S3 urls that are accessible
2 parents 7c79470 + b5e9e28 commit a0090e8

File tree

2 files changed

+7
-6
lines changed

2 files changed

+7
-6
lines changed

nucleus/dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ def annotate(
102102
self,
103103
annotations: List[Union[BoxAnnotation, PolygonAnnotation]],
104104
update: Optional[bool] = DEFAULT_ANNOTATION_UPDATE_MODE,
105-
batch_size: int = 20,
105+
batch_size: int = 5000,
106106
) -> dict:
107107
"""
108108
Uploads ground truth annotations for a given dataset.

tests/helpers.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -17,11 +17,11 @@
1717
TEST_DATASET_NAME = "[PyTest] Test Dataset"
1818
TEST_SLICE_NAME = "[PyTest] Test Slice"
1919
TEST_IMG_URLS = [
20-
"s3://scaleapi-attachments/BDD/BDD/bdd100k/images/100k/train/6dd63871-831611a6.jpg",
21-
"s3://scaleapi-attachments/BDD/BDD/bdd100k/images/100k/train/82c1005c-e2d1d94f.jpg",
22-
"s3://scaleapi-attachments/BDD/BDD/bdd100k/images/100k/train/7f2e1814-6591087d.jpg",
23-
"s3://scaleapi-attachments/BDD/BDD/bdd100k/images/100k/train/06924f46-1708b96f.jpg",
24-
"s3://scaleapi-attachments/BDD/BDD/bdd100k/images/100k/train/89b42832-10d662f4.jpg",
20+
"s3://scaleapi-cust-lidar/Hesai/raw_data/2019-5-11/hesai_data_1557540003/undistorted/front_camera/1557540143.650423lf.jpg",
21+
"s3://scaleapi-cust-lidar/Hesai/raw_data/2019-5-11/hesai_data_1557540003/undistorted/back_camera/1557540143.600352lf.jpg",
22+
"s3://scaleapi-cust-lidar/Hesai/raw_data/2019-5-11/hesai_data_1557540003/undistorted/right_camera/1557540143.681730lf.jpg",
23+
"s3://scaleapi-cust-lidar/Hesai/raw_data/2019-5-11/hesai_data_1557540003/undistorted/front_left_camera/1557540143.639619lf.jpg",
24+
"s3://scaleapi-cust-lidar/Hesai/raw_data/2019-5-11/hesai_data_1557540003/undistorted/front_right_camera/1557540143.661212lf.jpg",
2525
]
2626
TEST_DATASET_ITEMS = [
2727
DatasetItem(TEST_IMG_URLS[0], "1"),
@@ -62,6 +62,7 @@ def s3_sign(bucket, key):
6262
ExpiresIn=PRESIGN_EXPIRY_SECONDS,
6363
)
6464

65+
6566
def reference_id_from_url(url):
6667
return Path(url).name
6768

0 commit comments

Comments
 (0)