Skip to content

Commit bf30d6e

Browse files
authored
Bump-up Datumaro 1.2.0rc3 --> 1.3.2 and resolve errors (#2270)
Bump-up Datumaro and fix bugs
1 parent e41cb33 commit bf30d6e

11 files changed

+6
-6
lines changed

requirements/base.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ natsort>=6.0.0
44
prettytable
55
protobuf>=3.20.0
66
pyyaml
7-
datumaro==1.2.0rc3
7+
datumaro==1.3.2
88
psutil
99
scipy>=1.8
1010
bayesian-optimization>=1.2.0

tests/unit/core/data/adapter/test_init.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,7 @@ def test_direct_annotation(task_name, train_type):
105105
train_ann_files="tests/assets/car_tree_bug/annotations/instances_train_5_imgs.json",
106106
val_data_roots=os.path.join(root_path, data_root["val"]),
107107
)
108-
assert t_adapter.dataset[Subset.TRAINING].get_subset("train").get_annotated_items() == 5
108+
assert t_adapter.dataset[Subset.TRAINING].get_subset("train_5_imgs").get_annotated_items() == 5
109109

110110
v_adapter = get_dataset_adapter(
111111
task_type=task_type,
@@ -114,7 +114,7 @@ def test_direct_annotation(task_name, train_type):
114114
val_data_roots=os.path.join(root_path, data_root["val"]),
115115
val_ann_files="tests/assets/car_tree_bug/annotations/instances_val_1_imgs.json",
116116
)
117-
assert v_adapter.dataset[Subset.VALIDATION].get_subset("val").get_annotated_items() == 1
117+
assert v_adapter.dataset[Subset.VALIDATION].get_subset("val_1_imgs").get_annotated_items() == 1
118118

119119
tv_adapter = get_dataset_adapter(
120120
task_type=task_type,
@@ -124,8 +124,8 @@ def test_direct_annotation(task_name, train_type):
124124
val_data_roots=os.path.join(root_path, data_root["val"]),
125125
val_ann_files="tests/assets/car_tree_bug/annotations/instances_val_1_imgs.json",
126126
)
127-
assert tv_adapter.dataset[Subset.TRAINING].get_subset("train").get_annotated_items() == 5
128-
assert tv_adapter.dataset[Subset.VALIDATION].get_subset("val").get_annotated_items() == 1
127+
assert tv_adapter.dataset[Subset.TRAINING].get_subset("train_5_imgs").get_annotated_items() == 5
128+
assert tv_adapter.dataset[Subset.VALIDATION].get_subset("val_1_imgs").get_annotated_items() == 1
129129

130130

131131
@e2e_pytest_unit

tests/unit/core/data/manager/test_dataset_manager.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ def test_import_dataset_with_direct_annotation(self, data_root):
106106

107107
ann_files = "tests/assets/car_tree_bug/annotations/instances_train_5_imgs.json"
108108
train_dataset = DatasetManager.import_dataset(ann_files, data_format=data_format, subset="train")
109-
assert train_dataset.get_subset("train").get_annotated_items() == 5
109+
assert train_dataset.get_subset("train_5_imgs").get_annotated_items() == 5
110110

111111
@e2e_pytest_unit
112112
@pytest.mark.parametrize("task", AVAILABLE_TASKS)

0 commit comments

Comments
 (0)