Rahul-8799 commited on
Commit
8b2be02
·
verified ·
1 Parent(s): 661c978

Create utils/zip_output.py

Browse files
Files changed (1) hide show
  1. utils/zip_output.py +10 -0
utils/zip_output.py ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
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, _, files in os.walk(output_dir):
7
+ for file in files:
8
+ filepath = os.path.join(root, file)
9
+ zipf.write(filepath, os.path.relpath(filepath, output_dir))
10
+ return zip_name