File tree Expand file tree Collapse file tree 1 file changed +9
-6
lines changed Expand file tree Collapse file tree 1 file changed +9
-6
lines changed Original file line number Diff line number Diff line change @@ -69,14 +69,17 @@ def zip_files(
69
69
For Viya 3.5 models, ignore score code that is already in place in the file
70
70
directory provided. Default value is False.
71
71
"""
72
+
72
73
if isinstance (model_files , dict ):
73
- zip_buffer = io .BytesIO ()
74
- with zipfile .ZipFile (
75
- zip_buffer , "a" , zipfile .ZIP_DEFLATED , False
76
- ) as zip_file :
74
+ buffer = io .BytesIO ()
75
+
76
+ with zipfile .ZipFile (buffer , "w" , zipfile .ZIP_DEFLATED , False ) as archive :
77
77
for file_name , data in model_files .items ():
78
- zip_file .writestr (file_name , data )
79
- return io .BytesIO (zip_buffer .getvalue ())
78
+ archive .writestr (file_name , data )
79
+
80
+ # NOTE: bytes are added to the buffer when zip file is closed
81
+ # so ensure closed before returning
82
+ return buffer
80
83
else :
81
84
file_names = _filter_files (model_files , is_viya4 )
82
85
with zipfile .ZipFile (
You can’t perform that action at this time.
0 commit comments