minor fix to move zipped files into separate folders for part1, 2 and 3

lord2cc · Oct 30, 2017 · 899de96 · 899de96
1 parent ed654fc
commit 899de96
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/data/PROMISE12/setup.py b/data/PROMISE12/setup.py
@@ -11,8 +11,8 @@
 
 zip_dir = '.'
 target_dir = '.'
-for zip_filename in {'TrainingData_Part1.zip', 'TrainingData_Part1.zip',
-                     'TrainingData_Part1.zip'}:
+for zip_filename in {'TrainingData_Part1.zip', 'TrainingData_Part2.zip',
+                     'TrainingData_Part3.zip'}:
     zip_ref = zipfile.ZipFile(os.path.join(zip_dir, zip_filename), 'r')
-    zip_ref.extractall(target_dir)
-    zip_ref.close()
+    zip_ref.extractall(os.path.join(target_dir,zip_filename[:-4]))
+    zip_ref.close()