From a507c60eb977593b7c463f9d7d050b46bd0b2602 Mon Sep 17 00:00:00 2001
From: Jan Zavadil <79144013+hzavadil98@users.noreply.github.com>
Date: Thu, 20 Feb 2025 12:25:45 +0100
Subject: [PATCH] Fixing the downloader for MNIST closes#86

---
 CollaborativeCoding/dataloaders/download.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/CollaborativeCoding/dataloaders/download.py b/CollaborativeCoding/dataloaders/download.py
index 5e90beb..7681a7b 100644
--- a/CollaborativeCoding/dataloaders/download.py
+++ b/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():