nshtrainer 0.36.0__tar.gz → 0.37.0__tar.gz

{nshtrainer-0.36.0 → nshtrainer-0.37.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nshtrainer
-Version: 0.36.0
+Version: 0.37.0
 Summary: 
 Author: Nima Shoghi
 Author-email: nimashoghi@gmail.com

{nshtrainer-0.36.0 → nshtrainer-0.37.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nshtrainer"
-version = "0.36.0"
+version = "0.37.0"
 description = ""
 authors = ["Nima Shoghi <nimashoghi@gmail.com>"]
 readme = "README.md"

{nshtrainer-0.36.0 → nshtrainer-0.37.0}/src/nshtrainer/_checkpoint/saver.py

@@ -30,6 +30,8 @@ def _link_checkpoint(
                     linkpath.unlink(missing_ok=True)
         except Exception:
             log.warning(f"Failed to remove {linkpath}", exc_info=True)
+        else:
+            log.debug(f"Removed {linkpath=}")
 
         if metadata:
             _remove_checkpoint_metadata(linkpath)

{nshtrainer-0.36.0 → nshtrainer-0.37.0}/src/nshtrainer/util/path.py

@@ -80,11 +80,25 @@ def try_symlink_or_copy(
     link_path: Path,
     target_is_directory: bool = False,
     relative: bool = True,
+    remove_existing: bool = True,
 ):
     """
     Symlinks on Unix, copies on Windows.
     """
 
+    # If the link already exists, remove it
+    if remove_existing:
+        try:
+            if link_path.exists():
+                if link_path.is_dir():
+                    shutil.rmtree(link_path)
+                else:
+                    link_path.unlink(missing_ok=True)
+        except Exception:
+            log.warning(f"Failed to remove {link_path}", exc_info=True)
+        else:
+            log.debug(f"Removed {link_path=}")
+
     symlink_target = get_relative_path(link_path, file_path) if relative else file_path
     try:
         if platform.system() == "Windows":