fix: add dataset revision tag as recent version is broken

encord-team · Mar 11, 2024 · 25de2bd · 25de2bd
1 parent 0b130dd
commit 25de2bd
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/clip_eval/dataset/provider.py b/clip_eval/dataset/provider.py
@@ -41,7 +41,12 @@ def list_dataset_names(self) -> list[str]:
 # Add global settings
 dataset_provider.add_global_setting("cache_dir", CACHE_PATH)
 
-dataset_provider.register_dataset("LungCancer4Types", HFDataset, title_in_source="Kabil007/LungCancer4Types")
+dataset_provider.register_dataset(
+    "LungCancer4Types",
+    HFDataset,
+    title_in_source="Kabil007/LungCancer4Types",
+    revision="a1aab924c6bed6b080fc85552fd7b39724931605",
+)
 dataset_provider.register_dataset("plants", HFDataset, title_in_source="sampath017/plants")
 dataset_provider.register_dataset(
     "NIH-Chest-X-ray",