Merge pull request #18 from grok-ai/develop

Version 0.0.7

setup.cfg

@@ -6,7 +6,7 @@ author = Valentino Maiorca
 author_email = [email protected]
 keywords = python, torch
 license = MIT License
-url = https://github.com/Flegyas/nn-template-core
+url = https://github.com/grok-ai/nn-template-core
 
 [options]
 zip_safe = False

src/nn_core/serialization.py

@@ -95,6 +95,13 @@ def compress_checkpoint(src_dir: Path, dst_file: Path, delete_dir: bool = True):
     with zipfile.ZipFile(_normalize_path(dst_file), "w") as zip_file:
         for folder, subfolders, files in os.walk(src_dir):
             folder: Path = Path(folder)
+            for subfolder in subfolders:
+                zip_file.write(
+                    folder / subfolder,
+                    (folder / subfolder).relative_to(src_dir),
+                    compress_type=zipfile.ZIP_DEFLATED,
+                )
+
             for file in files:
                 zip_file.write(
                     folder / file,