fix adding files to zip

rochemedia · Feb 1, 2024 · a137398 · a137398
1 parent 340df38
commit a137398
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/llm_studio/app_utils/sections/experiment.py b/llm_studio/app_utils/sections/experiment.py
@@ -1683,7 +1683,7 @@ async def experiment_download_model(q: Q):
                 and file_path not in paths_added
                 and file_path != zip_path
             ):
-                add_file_to_zip(zf=zf, path=file)
+                add_file_to_zip(zf=zf, path=file_path)
                 paths_added.append(file_path)
                 logger.info(
                     f"Added {file_path} to zip file as it "

diff --git a/llm_studio/src/utils/utils.py b/llm_studio/src/utils/utils.py
@@ -110,7 +110,7 @@ def add_file_to_zip(zf: zipfile.ZipFile, path: str) -> None:
     try:
         zf.write(path, os.path.basename(path))
     except Exception:
-        pass
+        logger.warning(f"File {path} could not be added to zip.")
 
 
 def save_pickle(path: str, obj: Any, protocol: int = 4) -> None: