kostyl-toolkit 0.1.9__tar.gz → 0.1.10__tar.gz

{kostyl_toolkit-0.1.9 → kostyl_toolkit-0.1.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: kostyl-toolkit
-Version: 0.1.9
+Version: 0.1.10
 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.9 → kostyl_toolkit-0.1.10}/kostyl/ml_core/clearml/pulling_utils.py

@@ -1,4 +1,5 @@
 from pathlib import Path
+from typing import cast
 
 from clearml import InputModel
 from clearml import Task
@@ -47,7 +48,7 @@ def get_model_from_clearml[
     model: type[TModel],
     task: Task | None = None,
     ignore_remote_overrides: bool = True,
-) -> PreTrainedModel:
+) -> TModel:
     """
     Retrieve a pretrained model from ClearML and instantiate it using the appropriate loader.
 
@@ -83,4 +84,5 @@ def get_model_from_clearml[
             f"Unsupported model format for path: {local_path}. "
             "Expected a ClearML package directory or a .ckpt file."
         )
+    model_instance = cast(TModel, model_instance)
     return model_instance

{kostyl_toolkit-0.1.9 → kostyl_toolkit-0.1.10}/pyproject.toml

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