Merge branch 'main' into christian/update-dataloader-recall

Author: c-salomonsen

CollaborativeCoding/dataloaders/download.py

@@ -56,6 +56,7 @@ def _chech_is_downloaded(path: Path) -> bool:
                 return False
 
         def _download_data(path: Path) -> None:
+            path = path / "MNIST"
             urls = {key: MNIST_SOURCE[key][0] for key in MNIST_SOURCE.keys()}
 
             for name, url in urls.items():

tests/test_dataloaders.py

@@ -36,4 +36,4 @@ def test_load_data(data_name, expected):
     assert isinstance(dataset[0][0], torch.Tensor)
     assert isinstance(
         dataset[0][1], (int, torch.Tensor, np.ndarray)
-    )  # Should probably restrict this to only int or one-hot encoded tensor or array for consistency.
+    )  # Should probably restrict this to only int or one-hot encoded tensor or array for consistency.