kostyl-toolkit 0.1.32__tar.gz → 0.1.33__tar.gz

{kostyl_toolkit-0.1.32 → kostyl_toolkit-0.1.33}/PKG-INFO

@@ -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 → kostyl_toolkit-0.1.33}/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_toolkit-0.1.32 → kostyl_toolkit-0.1.33}/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 → kostyl_toolkit-0.1.33}/pyproject.toml

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