Merge pull request #127 from sofiia-chorna/fix-small-muad-url

🐛 Fix small MUAD dataset URL
ENSTA-U2IS-AI · Dec 26, 2024 · ec2d921 · ec2d921
2 parents 63e9ece + 5e5635b
commit ec2d921
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/tests/datamodules/segmentation/test_muad.py b/tests/datamodules/segmentation/test_muad.py
@@ -35,3 +35,7 @@ def test_camvid_main(self):
         dm.setup()
         dm.train_dataloader()
         dm.val_dataloader()
+
+    def test_small_muad_accessibility(self):
+        dataset = MUAD(root="./data/", split="test", version="small", download=True)
+        assert len(dataset.samples) > 0, "Dataset is not found"
diff --git a/torch_uncertainty/datasets/muad.py b/torch_uncertainty/datasets/muad.py
@@ -41,7 +41,7 @@ class MUAD(VisionDataset):
         "val_depth": "0282030d281aeffee3335f713ba12373",
     }
 
-    small_muad_url = "fira7s/Muad2"
+    small_muad_url = "ENSTA-U2IS/miniMUAD"
 
     _num_samples = {
         "full": {