Minor fix to check for zipfile (#437)

constantinpape · Dec 14, 2024 · 7517563 · 7517563
1 parent 082698f
commit 7517563
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/torch_em/data/datasets/histopathology/cpm.py b/torch_em/data/datasets/histopathology/cpm.py
@@ -61,7 +61,13 @@ def get_cpm_data(
         )
 
     os.makedirs(path, exist_ok=True)
-    zip_path = glob(os.path.join(path, f"{data_choice}*.zip"))[0]
+    zip_path = glob(os.path.join(path, f"{data_choice}*.zip"))
+    if len(zip_path) == 0:
+        raise AssertionError(
+            f"zip file for '{data_choice}' dataset is not found. Please download it from '{URL[data_choice]}'."
+        )
+
+    zip_path = zip_path[0]
     util.unzip(zip_path=zip_path, dst=path, remove=False)
 
     return data_dir