Dimitri Korsch před 4 roky
rodič
revize
aac8a33cc7
1 změnil soubory, kde provedl 4 přidání a 3 odebrání
  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)