kostyl-toolkit 0.1.24__tar.gz → 0.1.25__tar.gz

{kostyl_toolkit-0.1.24 → kostyl_toolkit-0.1.25}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: kostyl-toolkit
-Version: 0.1.24
+Version: 0.1.25
 Summary: Kickass Orchestration System for Training, Yielding & Logging 
 Requires-Dist: case-converter>=1.2.0
 Requires-Dist: loguru>=0.7.3

{kostyl_toolkit-0.1.24 → kostyl_toolkit-0.1.25}/kostyl/ml/lightning/callbacks/checkpoint.py

@@ -2,7 +2,6 @@ from datetime import timedelta
 from pathlib import Path
 from shutil import rmtree
 from typing import Literal
-from typing import cast
 
 from lightning.fabric.utilities.types import _PATH
 from lightning.pytorch.callbacks import ModelCheckpoint
@@ -244,6 +243,8 @@ class CustomModelCheckpoint(ModelCheckpoint):
         enable_version_counter: bool = True,
         registry_uploader_callback: RegistryUploaderCallback | None = None,
     ) -> None:
+        self.registry_uploader_callback = registry_uploader_callback
+        self._custom_best_model_path = ""
         super().__init__(
             dirpath=dirpath,
             filename=filename,
@@ -261,8 +262,6 @@ class CustomModelCheckpoint(ModelCheckpoint):
             save_on_train_epoch_end=save_on_train_epoch_end,
             enable_version_counter=enable_version_counter,
         )
-        self.registry_uploader_callback = registry_uploader_callback
-        self._custom_best_model_path = cast(str, self.best_model_path)
         return
 
     @property

{kostyl_toolkit-0.1.24 → kostyl_toolkit-0.1.25}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "kostyl-toolkit"
-version = "0.1.24"
+version = "0.1.25"
 description = "Kickass Orchestration System for Training, Yielding & Logging "
 readme = "README.md"
 requires-python = ">=3.12"