Rahul-8799 commited on
Commit
e7aecda
·
verified ·
1 Parent(s): 9e8fc5f

Update utils/zip_output.py

Browse files
Files changed (1) hide show
  1. utils/zip_output.py +10 -5
utils/zip_output.py CHANGED
@@ -1,9 +1,14 @@
1
  import zipfile
2
  import os
3
 
4
- def zip_output(output_dir="output", zip_name="output_bundle.zip"):
5
- with zipfile.ZipFile(zip_name, 'w') as zipf:
6
- for root, dirs, files in os.walk(output_dir):
 
 
 
7
  for file in files:
8
- zipf.write(os.path.join(root, file), file)
9
- return zip_name
 
 
 
1
  import zipfile
2
  import os
3
 
4
+ def zip_output():
5
+ zip_filename = "output_bundle.zip"
6
+ zip_path = f"output/{zip_filename}"
7
+
8
+ with zipfile.ZipFile(zip_path, "w") as zipf:
9
+ for folder, _, files in os.walk("output"):
10
  for file in files:
11
+ if file != zip_filename:
12
+ zipf.write(os.path.join(folder, file), arcname=file)
13
+
14
+ return zip_path