kostyl-toolkit 0.1.13__py3-none-any.whl → 0.1.14__py3-none-any.whl

kostyl/ml/lightning/extenstions/custom_module.py

@@ -26,8 +26,6 @@ logger = setup_logger(fmt="only_message")
 class KostylLightningModule(L.LightningModule):
     """Custom PyTorch Lightning Module with logging, checkpointing, and distributed training utilities."""
 
-    model: PreTrainedModel | nn.Module | None
-
     def get_process_group(self) -> ProcessGroup | None:
         """
         Retrieves the data parallel process group for distributed training.
@@ -54,16 +52,15 @@ class KostylLightningModule(L.LightningModule):
             dp_pg = dist.group.WORLD
         return dp_pg
 
-    def get_model(self) -> PreTrainedModel | nn.Module:
+    @property
+    def model_instance(self) -> PreTrainedModel | nn.Module:
         """Returns the underlying model."""
-        if self.model is None:
-            raise ValueError("Model is not configured.")
-        return self.model
+        raise NotImplementedError
 
     @property
     def model_config(self) -> PretrainedConfig | None:
         """Returns the model configuration if available."""
-        model = self.get_model()
+        model = self.model_instance
         if hasattr(model, "config"):
             return model.config  # type: ignore
         return None
@@ -75,7 +72,7 @@ class KostylLightningModule(L.LightningModule):
 
     @override
     def on_save_checkpoint(self, checkpoint: dict[str, Any]) -> None:
-        model = self.get_model()
+        model = self.model_instance
         if hasattr(model, "config"):
             cfg = model.config
             if hasattr(cfg, "to_dict"):
@@ -94,9 +91,6 @@ class KostylLightningModule(L.LightningModule):
 
     @override
     def on_before_optimizer_step(self, optimizer) -> None:
-        if self.model is None:
-            raise ValueError("Model must be configured before optimizer step.")
-
         grad_clip_val = self.grad_clip_val
         if grad_clip_val is None:
             return

{kostyl_toolkit-0.1.13.dist-info → kostyl_toolkit-0.1.14.dist-info}/METADATA

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

@@ -15,7 +15,7 @@ kostyl/ml/lightning/callbacks/checkpoint.py,sha256=FooGeeUz6TtoXQglpcK16NWAmSX3f
 kostyl/ml/lightning/callbacks/early_stopping.py,sha256=D5nyjktCJ9XYAf28-kgXG8jORvXLl1N3nbDQnvValPM,615
 kostyl/ml/lightning/callbacks/registry_uploading.py,sha256=um-fHQUp-TSD_76rVZoNV2IMi05RFjKajGvMV7SAb6s,4691
 kostyl/ml/lightning/extenstions/__init__.py,sha256=OY6QGv1agYgqqKf1xJBrxgp_i8FunVfPzYezfaRrGXU,182
-kostyl/ml/lightning/extenstions/custom_module.py,sha256=N4kPObp4jqrCJscPKzJib3AkvcQQ3AHal7A1Cm3ZobA,6201
+kostyl/ml/lightning/extenstions/custom_module.py,sha256=nB5jW7cqRD1tyh-q5LD2EtiFQwFkLXpnS9Yu6c5xMRg,5987
 kostyl/ml/lightning/extenstions/pretrained_model.py,sha256=ZOKtrVl095cwvI43wAz-Xdzu4l0v0lHH2mfh4WXwxKQ,5059
 kostyl/ml/lightning/loggers/__init__.py,sha256=e51dszaoJbuzwBkbdugmuDsPldoSO4yaRgmZUg1Bdy0,71
 kostyl/ml/lightning/loggers/tb_logger.py,sha256=I0WQV3SVVIds9kHhAYnN3SXodX64jfTk7_5gH-j3uzA,932
@@ -30,6 +30,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=3MvfDPArZhwakHu5nMlp_LpOsWg0E0SP26y41clsBtA,5232
-kostyl_toolkit-0.1.13.dist-info/WHEEL,sha256=3id4o64OvRm9dUknh3mMJNcfoTRK08ua5cU6DFyVy-4,79
-kostyl_toolkit-0.1.13.dist-info/METADATA,sha256=BX_FpC83Q61Mr4jBEjRj-xC5UGtZcQJJn3EoF-Dnx24,4269
-kostyl_toolkit-0.1.13.dist-info/RECORD,,
+kostyl_toolkit-0.1.14.dist-info/WHEEL,sha256=3id4o64OvRm9dUknh3mMJNcfoTRK08ua5cU6DFyVy-4,79
+kostyl_toolkit-0.1.14.dist-info/METADATA,sha256=lOiHm_Bd3uf2eHv7unCgagPABhJUF793ZO3FtX5PWNc,4269
+kostyl_toolkit-0.1.14.dist-info/RECORD,,