@@ -105,7 +105,7 @@ def test_direct_annotation(task_name, train_type):
105
105
train_ann_files = "tests/assets/car_tree_bug/annotations/instances_train_5_imgs.json" ,
106
106
val_data_roots = os .path .join (root_path , data_root ["val" ]),
107
107
)
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
109
109
110
110
v_adapter = get_dataset_adapter (
111
111
task_type = task_type ,
@@ -114,7 +114,7 @@ def test_direct_annotation(task_name, train_type):
114
114
val_data_roots = os .path .join (root_path , data_root ["val" ]),
115
115
val_ann_files = "tests/assets/car_tree_bug/annotations/instances_val_1_imgs.json" ,
116
116
)
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
118
118
119
119
tv_adapter = get_dataset_adapter (
120
120
task_type = task_type ,
@@ -124,8 +124,8 @@ def test_direct_annotation(task_name, train_type):
124
124
val_data_roots = os .path .join (root_path , data_root ["val" ]),
125
125
val_ann_files = "tests/assets/car_tree_bug/annotations/instances_val_1_imgs.json" ,
126
126
)
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
129
129
130
130
131
131
@e2e_pytest_unit
0 commit comments