cleanup

Author: pmeier

test/builtin_dataset_mocks.py

@@ -1212,7 +1212,7 @@ def _make_segmentations(cls, root, image_files):
                 size=[1, *make_tensor((2,), low=3, dtype=torch.int).tolist()],
             )
 
-        make_tar(root, segmentations_folder.with_suffix(".tgz").name)
+        make_tar(root, segmentations_folder.with_suffix(".tgz").name, compression="gz")
 
     @classmethod
     def generate(cls, root):

test/test_prototype_builtin_datasets.py

@@ -116,6 +116,7 @@ def scan(graph):
         else:
             raise AssertionError(f"The dataset doesn't comprise a {annotation_dp_type.__name__}() datapipe.")
 
+    @parametrize_dataset_mocks(DATASET_MOCKS)
     def test_loadable_through_api(self, mocker, dataset_mock, config):
         # Make all resources that are necessary for the given config
         dataset_mock.make_mock_resources(config)