|
@@ -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)
|
|
|
|