Dimitri Korsch 4 年 前
コミット
aac8a33cc7
1 ファイル変更4 行追加3 行削除
  1. 4 3
      tests/test_annotations.py

+ 4 - 3
tests/test_annotations.py

@@ -13,7 +13,6 @@ from cvdatasets import FileListAnnotations
 from cvdatasets.utils import read_info_file
 
 class MockAnnotation(FileListAnnotations):
-	name = "MOCK"
 	index_offset = 0
 
 	@property
@@ -107,7 +106,8 @@ class AnnotationTest(BaseAnnotationTest):
 			labels=[i % 5 for i in range(10)],
 			split=[int(i < 5) for i in range(10)],
 			bboxes=True,
-			n_parts=5
+			n_parts=5,
+			annot_params=dict(dataset_key="MOCK")
 		)
 		annot = self.create_annotations(**_annotation_params)
 
@@ -118,7 +118,8 @@ class AnnotationTest(BaseAnnotationTest):
 			labels=[i % 5 for i in range(10)],
 			split=[int(i < 5) for i in range(10)],
 			bboxes=True,
-			n_parts=5
+			n_parts=5,
+			annot_params=dict(dataset_key="MOCK")
 		)
 		annot = self.create_annotations(**_annotation_params)