kostyl-toolkit 0.1.24__py3-none-any.whl → 0.1.25__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (4) hide show
  1. kostyl/ml/lightning/callbacks/checkpoint.py +2 -3
  2. {kostyl_toolkit-0.1.24.dist-info → kostyl_toolkit-0.1.25.dist-info}/METADATA +1 -1
  3. {kostyl_toolkit-0.1.24.dist-info → kostyl_toolkit-0.1.25.dist-info}/RECORD +4 -4
  4. {kostyl_toolkit-0.1.24.dist-info → kostyl_toolkit-0.1.25.dist-info}/WHEEL +0 -0
kostyl/ml/lightning/callbacks/checkpoint.py CHANGED
@@ -2,7 +2,6 @@ from datetime import timedelta
2
2
  from pathlib import Path
3
3
  from shutil import rmtree
4
4
  from typing import Literal
5
- from typing import cast
6
5
 
7
6
  from lightning.fabric.utilities.types import _PATH
8
7
  from lightning.pytorch.callbacks import ModelCheckpoint
@@ -244,6 +243,8 @@ class CustomModelCheckpoint(ModelCheckpoint):
244
243
  enable_version_counter: bool = True,
245
244
  registry_uploader_callback: RegistryUploaderCallback | None = None,
246
245
  ) -> None:
246
+ self.registry_uploader_callback = registry_uploader_callback
247
+ self._custom_best_model_path = ""
247
248
  super().__init__(
248
249
  dirpath=dirpath,
249
250
  filename=filename,
@@ -261,8 +262,6 @@ class CustomModelCheckpoint(ModelCheckpoint):
261
262
  save_on_train_epoch_end=save_on_train_epoch_end,
262
263
  enable_version_counter=enable_version_counter,
263
264
  )
264
- self.registry_uploader_callback = registry_uploader_callback
265
- self._custom_best_model_path = cast(str, self.best_model_path)
266
265
  return
267
266
 
268
267
  @property
{kostyl_toolkit-0.1.24.dist-info → kostyl_toolkit-0.1.25.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: kostyl-toolkit
3
- Version: 0.1.24
3
+ Version: 0.1.25
4
4
  Summary: Kickass Orchestration System for Training, Yielding & Logging
5
5
  Requires-Dist: case-converter>=1.2.0
6
6
  Requires-Dist: loguru>=0.7.3
{kostyl_toolkit-0.1.24.dist-info → kostyl_toolkit-0.1.25.dist-info}/RECORD RENAMED
@@ -11,7 +11,7 @@ kostyl/ml/configs/training_settings.py,sha256=Sq2tiRuwkbmi9zKDG2JghZLXo5DDt_eQqN
11
11
  kostyl/ml/dist_utils.py,sha256=Onf0KHVLA8oeUgZTcTdmR9qiM22f2uYLoNwgLbMGJWk,3495
12
12
  kostyl/ml/lightning/__init__.py,sha256=-F3JAyq8KU1d-nACWryGu8d1CbvWbQ1rXFdeRwfE2X8,175
13
13
  kostyl/ml/lightning/callbacks/__init__.py,sha256=enexQt3octktsTiEYHltSF_24CM-NeFEVFimXiavGiY,296
14
- kostyl/ml/lightning/callbacks/checkpoint.py,sha256=rJ05S7BnPopvQjV5b9CI29-S7-ySllyhG7PIzz34VyY,16793
14
+ kostyl/ml/lightning/callbacks/checkpoint.py,sha256=1gk5-NjsMXe5cZP0OgNcoc9KUTzRDTHIokVEDr74sjI,16740
15
15
  kostyl/ml/lightning/callbacks/early_stopping.py,sha256=D5nyjktCJ9XYAf28-kgXG8jORvXLl1N3nbDQnvValPM,615
16
16
  kostyl/ml/lightning/callbacks/registry_uploader.py,sha256=ksoh02dzIde4E_GaZykfiOgfSjZti-IJt_i61enem3s,6779
17
17
  kostyl/ml/lightning/extenstions/__init__.py,sha256=OY6QGv1agYgqqKf1xJBrxgp_i8FunVfPzYezfaRrGXU,182
@@ -30,6 +30,6 @@ kostyl/utils/__init__.py,sha256=hkpmB6c5pr4Ti5BshOROebb7cvjDZfNCw83qZ_FFKMM,240
30
30
  kostyl/utils/dict_manipulations.py,sha256=e3vBicID74nYP8lHkVTQc4-IQwoJimrbFELy5uSF6Gk,1073
31
31
  kostyl/utils/fs.py,sha256=gAQNIU4R_2DhwjgzOS8BOMe0gZymtY1eZwmdgOdDgqo,510
32
32
  kostyl/utils/logging.py,sha256=Vye0u4-yeOSUc-f03gpQbxSktTbFiilTWLEVr00ZHvc,5796
33
- kostyl_toolkit-0.1.24.dist-info/WHEEL,sha256=z-mOpxbJHqy3cq6SvUThBZdaLGFZzdZPtgWLcP2NKjQ,79
34
- kostyl_toolkit-0.1.24.dist-info/METADATA,sha256=uq8MPJ9vJgWsp9Z2c7C9tcbaH29QM9ux7_SyahPSlHE,4269
35
- kostyl_toolkit-0.1.24.dist-info/RECORD,,
33
+ kostyl_toolkit-0.1.25.dist-info/WHEEL,sha256=z-mOpxbJHqy3cq6SvUThBZdaLGFZzdZPtgWLcP2NKjQ,79
34
+ kostyl_toolkit-0.1.25.dist-info/METADATA,sha256=Go9dF8W4vQ4HzQpzRCLa-i9NsYq5o0J34NtvGWZnEvA,4269
35
+ kostyl_toolkit-0.1.25.dist-info/RECORD,,