kostyl-toolkit 0.1.27__tar.gz → 0.1.29__tar.gz

{kostyl_toolkit-0.1.27 → kostyl_toolkit-0.1.29}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: kostyl-toolkit
-Version: 0.1.27
+Version: 0.1.29
 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.27 → kostyl_toolkit-0.1.29}/kostyl/ml/lightning/callbacks/registry_uploader.py

@@ -69,6 +69,8 @@ class ClearMLRegistryUploaderCallback(RegistryUploaderCallback):
         self._output_model: OutputModel | None = None
         self._last_uploaded_model_path: str = ""
         self._upload_callback: Callable | None = None
+
+        self._design_updated: bool = False
         return
 
     def _create_output_model(self) -> OutputModel:
@@ -116,7 +118,9 @@ class ClearMLRegistryUploaderCallback(RegistryUploaderCallback):
             auto_delete_file=False,
             async_enable=False,
         )
-        self._output_model.update_design(config_dict=self.config_dict)
+        if not self._design_updated and self.config_dict is not None:
+            self._output_model.update_design(config_dict=self.config_dict)
+            self._design_updated = True
 
         self._last_uploaded_model_path = path
         return

{kostyl_toolkit-0.1.27 → kostyl_toolkit-0.1.29}/kostyl/ml/schedulers/linear.py

@@ -13,21 +13,21 @@ class _LinearScheduleBase(BaseScheduler):
         self,
         param_name: str,
         num_iters: int,
-        base_value: float,
+        start_value: float,
         final_value: float,
     ) -> None:
         self.param_name = param_name
         self.num_iters = num_iters
-        self.base_value = base_value
+        self.start_value = start_value
         self.final_value = final_value
 
         self.scheduler_values: npt.NDArray[np.float64] = np.array([], dtype=np.float64)
-        self.current_value_ = self.base_value
+        self.current_value_ = self.start_value
         return
 
     def _create_scheduler(self) -> None:
         self.scheduler_values = np.linspace(
-            self.base_value, self.final_value, num=self.num_iters, dtype=np.float64
+            self.start_value, self.final_value, num=self.num_iters, dtype=np.float64
         )
         if len(self.scheduler_values) != self.num_iters:
             raise ValueError(
@@ -69,7 +69,7 @@ class LinearScheduler(_LinearScheduleBase):
         optimizer: torch.optim.Optimizer,
         param_group_field: str,
         num_iters: int,
-        base_value: float,
+        start_value: float,
         final_value: float,
         multiplier_field: str | None = None,
         skip_if_zero: bool = False,
@@ -83,7 +83,7 @@ class LinearScheduler(_LinearScheduleBase):
             optimizer: Optimizer whose param groups are updated in-place.
             param_group_field: Name of the field that receives the scheduled value.
             num_iters: Number of scheduler iterations before clamping at ``final_value``.
-            base_value: Value used on the first iteration.
+            start_value: Value used on the first iteration.
             final_value: Value used once ``num_iters`` iterations are consumed.
             multiplier_field: Optional per-group multiplier applied to the scheduled value.
             skip_if_zero: Leave groups untouched when their target field equals zero.
@@ -99,7 +99,7 @@ class LinearScheduler(_LinearScheduleBase):
         super().__init__(
             param_name=param_group_field,
             num_iters=num_iters,
-            base_value=base_value,
+            start_value=start_value,
             final_value=final_value,
         )
         self.param_group_field = param_group_field

{kostyl_toolkit-0.1.27 → kostyl_toolkit-0.1.29}/pyproject.toml

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