kostyl-toolkit 0.1.32__py3-none-any.whl → 0.1.33__py3-none-any.whl

kostyl/ml/lightning/callbacks/checkpoint.py

@@ -301,7 +301,7 @@ def setup_checkpoint_callback(
     ckpt_cfg: CheckpointConfig,
     save_weights_only: bool = True,
     registry_uploader_callback: RegistryUploaderCallback | None = None,
-    uploading_mode: Literal["only-best", "every-checkpoint"] | None = None,
+    uploading_strategy: Literal["only-best", "every-checkpoint"] | None = None,
 ) -> ModelCheckpointWithRegistryUploader | ModelCheckpoint:
     """
     Create and configure a checkpoint callback for model saving.
@@ -316,8 +316,8 @@ def setup_checkpoint_callback(
         save_weights_only: If True, only model weights are saved without optimizer and lr-scheduler state.
             Defaults to True.
         registry_uploader_callback: Optional callback for uploading checkpoints to a remote registry.
-            Must be specified together with uploading_mode.
-        uploading_mode: Checkpoint upload mode:
+            Must be specified together with uploading_strategy.
+        uploading_strategy: Checkpoint upload mode:
             - "only-best": only the best checkpoint is uploaded
             - "every-checkpoint": every saved checkpoint is uploaded
             Must be specified together with registry_uploader_callback.
@@ -334,7 +334,7 @@ def setup_checkpoint_callback(
         (only on the main process in distributed training).
 
     """
-    if (registry_uploader_callback is None) != (uploading_mode is None):
+    if (registry_uploader_callback is None) != (uploading_strategy is None):
         raise ValueError(
             "Both registry_uploader_callback and uploading_mode must be provided or neither."
         )
@@ -348,7 +348,7 @@ def setup_checkpoint_callback(
         logger.info(f"Creating checkpoint directory {dirpath}.")
         dirpath.mkdir(parents=True, exist_ok=True)
 
-    if (registry_uploader_callback is not None) and (uploading_mode is not None):
+    if (registry_uploader_callback is not None) and (uploading_strategy is not None):
         checkpoint_callback = ModelCheckpointWithRegistryUploader(
             dirpath=dirpath,
             filename=ckpt_cfg.filename,
@@ -358,7 +358,7 @@ def setup_checkpoint_callback(
             verbose=True,
             save_weights_only=save_weights_only,
             registry_uploader_callback=registry_uploader_callback,
-            uploading_mode=uploading_mode,
+            uploading_mode=uploading_strategy,
         )
     else:
         checkpoint_callback = ModelCheckpoint(

kostyl/ml/lightning/training_utils.py

@@ -98,9 +98,9 @@ def setup_callbacks(
     task: Task,
     root_path: Path,
     checkpoint_cfg: CheckpointConfig,
-    uploading_mode: Literal["only-best", "every-checkpoint"],
+    early_stopping_cfg: EarlyStoppingConfig | None,
     output_model: OutputModel,
-    early_stopping_cfg: EarlyStoppingConfig | None = None,
+    checkpoint_upload_strategy: Literal["only-best", "every-checkpoint"],
     config_dict: dict[str, str] | None = None,
     enable_tag_versioning: bool = False,
 ) -> Callbacks:
@@ -114,7 +114,7 @@ def setup_callbacks(
         task: ClearML task for organizing checkpoints by task name and ID.
         root_path: Root directory for saving checkpoints.
         checkpoint_cfg: Configuration for checkpoint saving behavior.
-        uploading_mode: Model upload strategy:
+        checkpoint_upload_strategy: Model upload strategy:
             - `"only-best"`: Upload only the best checkpoint based on monitored metric.
             - `"every-checkpoint"`: Upload every saved checkpoint.
         output_model: ClearML OutputModel instance for model registry integration.
@@ -143,7 +143,7 @@ def setup_callbacks(
         root_path / "checkpoints" / task.name / task.id,
         checkpoint_cfg,
         registry_uploader_callback=model_uploader,
-        uploading_mode=uploading_mode,
+        uploading_strategy=checkpoint_upload_strategy,
     )
     if early_stopping_cfg is not None:
         early_stopping_callback = setup_early_stopping_callback(early_stopping_cfg)

{kostyl_toolkit-0.1.32.dist-info → kostyl_toolkit-0.1.33.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: kostyl-toolkit
-Version: 0.1.32
+Version: 0.1.33
 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.32.dist-info → kostyl_toolkit-0.1.33.dist-info}/RECORD

@@ -12,14 +12,14 @@ kostyl/ml/data_processing_utils.py,sha256=jjEjV0S0wREgZkzg27ip0LpI8cQqkwe2QwATmA
 kostyl/ml/dist_utils.py,sha256=Onf0KHVLA8oeUgZTcTdmR9qiM22f2uYLoNwgLbMGJWk,3495
 kostyl/ml/lightning/__init__.py,sha256=R36PImjVvzBF9t_z9u6RYVnUFJJ-sNDUOdboWUojHmM,173
 kostyl/ml/lightning/callbacks/__init__.py,sha256=EnKkNwwNDZnEqKRlpY4FVrqP88ECPF6nlT2bSLUIKRk,194
-kostyl/ml/lightning/callbacks/checkpoint.py,sha256=t3bfJH1HhWFA2tFArCaDe5BshFAaJ6-LRlQx5y5eOh8,18959
+kostyl/ml/lightning/callbacks/checkpoint.py,sha256=sZ9OqudO-gXp7FqtWaOH46TXVpeCJxV-EowyRPN836k,18983
 kostyl/ml/lightning/callbacks/early_stopping.py,sha256=D5nyjktCJ9XYAf28-kgXG8jORvXLl1N3nbDQnvValPM,615
 kostyl/ml/lightning/extensions/__init__.py,sha256=OY6QGv1agYgqqKf1xJBrxgp_i8FunVfPzYezfaRrGXU,182
 kostyl/ml/lightning/extensions/custom_module.py,sha256=iQrnPz-WTmRfvLo94C5fQc2Qwa1IpHtUh1sCpVwTSFM,6602
 kostyl/ml/lightning/extensions/pretrained_model.py,sha256=QJGr2UvYJcU2Gy2w8z_cEvTodjv7hGdd2PPPfdOI-Mw,4017
 kostyl/ml/lightning/loggers/__init__.py,sha256=e51dszaoJbuzwBkbdugmuDsPldoSO4yaRgmZUg1Bdy0,71
 kostyl/ml/lightning/loggers/tb_logger.py,sha256=j02HK5ue8yzXXV8FWKmmXyHkFlIxgHx-ahHWk_rFCZs,893
-kostyl/ml/lightning/training_utils.py,sha256=WSOMlqfNC8IRLl2llf9vKUTJwV8qgwWaEItft68VZSM,9019
+kostyl/ml/lightning/training_utils.py,sha256=u7X9ysF9Gqy8CdwacdcDlNQNsbagYAhslbv-1WLJ45k,9052
 kostyl/ml/metrics_formatting.py,sha256=U6vdNENZLvp2dT1L3HqFKtXrHwGKoDXN93hvamPGHjM,1341
 kostyl/ml/params_groups.py,sha256=nUyw5d06Pvy9QPiYtZzLYR87xwXqJLxbHthgQH8oSCM,3583
 kostyl/ml/registry_uploader.py,sha256=W90TYo_WKv2oBE6nqEJl4hecYmJyyuKwQJ9_uUPGnJQ,3346
@@ -32,6 +32,6 @@ kostyl/utils/__init__.py,sha256=hkpmB6c5pr4Ti5BshOROebb7cvjDZfNCw83qZ_FFKMM,240
 kostyl/utils/dict_manipulations.py,sha256=e3vBicID74nYP8lHkVTQc4-IQwoJimrbFELy5uSF6Gk,1073
 kostyl/utils/fs.py,sha256=gAQNIU4R_2DhwjgzOS8BOMe0gZymtY1eZwmdgOdDgqo,510
 kostyl/utils/logging.py,sha256=Vye0u4-yeOSUc-f03gpQbxSktTbFiilTWLEVr00ZHvc,5796
-kostyl_toolkit-0.1.32.dist-info/WHEEL,sha256=ZyFSCYkV2BrxH6-HRVRg3R9Fo7MALzer9KiPYqNxSbo,79
-kostyl_toolkit-0.1.32.dist-info/METADATA,sha256=Chx6XpZE5vULjb6b9k-8R044xdsjq8lhO182Wi8PJa4,4269
-kostyl_toolkit-0.1.32.dist-info/RECORD,,
+kostyl_toolkit-0.1.33.dist-info/WHEEL,sha256=ZyFSCYkV2BrxH6-HRVRg3R9Fo7MALzer9KiPYqNxSbo,79
+kostyl_toolkit-0.1.33.dist-info/METADATA,sha256=O-KYNVz8-H1ihBGjaqKLbaecroD3dx2guZ8hzR-M2Wg,4269
+kostyl_toolkit-0.1.33.dist-info/RECORD,,