simplify QMNIST download logic (#4562)

a485b8c6 · Philip Meier · GitHub · 05dcf50a · a485b8c6
Unverified Commit a485b8c6 authored Oct 07, 2021 by Philip Meier Committed by GitHub Oct 07, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 4 deletions

torchvision/datasets/mnist.py torchvision/datasets/mnist.py +1 -4

No files found.
--- a/torchvision/datasets/mnist.py
+++ b/torchvision/datasets/mnist.py
@@ -466,10 +466,7 @@ class QMNIST(MNIST):
        split = self.resources[self.subsets[self.what]]

        for url, md5 in split:
-            filename = url.rpartition("/")[2]
-            file_path = os.path.join(self.raw_folder, filename)
-            if not os.path.isfile(file_path):
-                download_and_extract_archive(url, self.raw_folder, filename=filename, md5=md5)
+            download_and_extract_archive(url, self.raw_folder, md5=md5)

    def __getitem__(self, index: int) -> Tuple[Any, Any]:
        # redefined to handle the compat flag