fix: unzip and delete zipped files

dimakis · Nov 20, 2023 · 1fce8b9 · 1fce8b9
1 parent 444ce4b
commit 1fce8b9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/demo-notebooks/guided-demos/download_mnist_datasets.py b/demo-notebooks/guided-demos/download_mnist_datasets.py
@@ -39,6 +39,10 @@ def download_mnist_dataset(destination_dir):
 
     print(f"MNIST dataset downloaded in {destination_dir}")
 
+    # unzip the gzipped files
+    [os.system(f"gunzip {destination_dir}/{file}") for file in os.listdir(destination_dir) if file.endswith(".gz")]
+    # delete the gzipped files
+    [os.remove(f"{destination_dir}/{file}") for file in os.listdir(destination_dir) if file.endswith(".gz")]
 
 # Specify the directory where you
 destination_dir = os.path.dirname(os.path.abspath(__file__))