nshtrainer 0.22.0__tar.gz → 0.22.1__tar.gz

{nshtrainer-0.22.0 → nshtrainer-0.22.1}/PKG-INFO

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

{nshtrainer-0.22.0 → nshtrainer-0.22.1}/pyproject.toml

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

{nshtrainer-0.22.0 → nshtrainer-0.22.1}/src/nshtrainer/_checkpoint/metadata.py

@@ -11,6 +11,7 @@ import numpy as np
 import torch
 
 from ..util._environment_info import EnvironmentConfig
+from ..util.path import get_relative_path
 
 if TYPE_CHECKING:
     from ..model import BaseConfig, LightningModuleBase
@@ -145,7 +146,7 @@ def _link_checkpoint_metadata(checkpoint_path: Path, linked_checkpoint_path: Pat
             # We should store the path as a relative path
             # to the metadata file to avoid issues with
             # moving the checkpoint directory
-            linked_path.symlink_to(path.relative_to(linked_path.parent))
+            linked_path.symlink_to(get_relative_path(linked_path, path))
         except OSError:
             # on Windows, special permissions are required to create symbolic links as a regular user
             # fall back to copying the file

{nshtrainer-0.22.0 → nshtrainer-0.22.1}/src/nshtrainer/trainer/trainer.py

@@ -439,7 +439,8 @@ class Trainer(LightningTrainer):
         ):
             # If we have a cached path, then we symlink it to the new path.
             log.info(f"Re-using cached checkpoint {cached_path} for {filepath}.")
-            _link_checkpoint(cached_path, filepath, metadata=False)
+            if self.is_global_zero:
+                _link_checkpoint(cached_path, filepath, metadata=False)
         else:
             super().save_checkpoint(filepath, weights_only, storage_options)