Chesapeake data uses proprietary compression

microsoft · Mar 19, 2022 · a543c24 · a543c24
1 parent 9dbc104
commit a543c24
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/trainers/test_byol.py b/tests/trainers/test_byol.py
@@ -43,6 +43,9 @@ class TestBYOLTask:
         ],
     )
     def test_trainer(self, name: str, classname: Type[LightningDataModule]) -> None:
+        if name == "chesapeake_cvpr_7":
+            pytest.importorskip("zipfile_deflate64")
+
         conf = OmegaConf.load(os.path.join("conf", "task_defaults", name + ".yaml"))
         conf_dict = OmegaConf.to_object(conf.experiment)
         conf_dict = cast(Dict[Any, Dict[Any, Any]], conf_dict)