Dimitri Korsch hace 4 años
padre
commit
aac8a33cc7
Se han modificado 1 ficheros con 4 adiciones y 3 borrados
  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)