truefoundry 0.4.6__py3-none-any.whl → 0.5.0rc2__py3-none-any.whl

truefoundry/ml/mlfoundry_run.py

@@ -20,7 +20,7 @@ from urllib.parse import urljoin, urlsplit
 
 from truefoundry import version
 from truefoundry.common.utils import relogin_error_message
-from truefoundry.ml import constants, enums
+from truefoundry.ml import constants
 from truefoundry.ml.autogen.client import (  # type: ignore[attr-defined]
     ArtifactType,
     DeleteRunRequest,
@@ -40,13 +40,17 @@ from truefoundry.ml.autogen.client import (  # type: ignore[attr-defined]
     UpdateRunRequestDto,
 )
 from truefoundry.ml.entities import Metric
-from truefoundry.ml.enums import RunStatus
+from truefoundry.ml.enums import ModelFramework, RunStatus
 from truefoundry.ml.exceptions import MlFoundryException
 from truefoundry.ml.internal_namespace import NAMESPACE
 from truefoundry.ml.log_types import Image, Plot
 from truefoundry.ml.log_types.artifacts.artifact import ArtifactPath, ArtifactVersion
 from truefoundry.ml.log_types.artifacts.general_artifact import _log_artifact_version
-from truefoundry.ml.log_types.artifacts.model import ModelVersion, _log_model_version
+from truefoundry.ml.log_types.artifacts.model import (
+    BlobStorageModelDirectory,
+    ModelVersion,
+    _log_model_version,
+)
 from truefoundry.ml.logger import logger
 from truefoundry.ml.run_utils import ParamsType, flatten_dict, process_params
 from truefoundry.ml.session import ACTIVE_RUNS, _get_api_client, get_active_session
@@ -61,6 +65,8 @@ if TYPE_CHECKING:
     import matplotlib
     import plotly
 
+    from truefoundry.ml import ModelFrameworkType
+
 
 def _ensure_not_deleted(method):
     @functools.wraps(method)
@@ -920,13 +926,13 @@ class MlFoundryRun:
         self,
         *,
         name: str,
-        model_file_or_folder: str,
-        framework: Optional[Union[enums.ModelFramework, str]],
+        model_file_or_folder: Union[str, BlobStorageModelDirectory],
         additional_files: Sequence[Tuple[Union[str, Path], Optional[str]]] = (),
         description: Optional[str] = None,
         metadata: Optional[Dict[str, Any]] = None,
         step: int = 0,
         progress: Optional[bool] = None,
+        framework: Optional[Union[str, ModelFramework, "ModelFrameworkType"]] = None,
     ) -> ModelVersion:
         # TODO (chiragjn): Document mapping of framework to list of valid model save kwargs
         # TODO (chiragjn): Add more examples
@@ -939,12 +945,21 @@ class MlFoundryRun:
             name (str): Name of the model. If a model with this name already exists under the current ML Repo,
                 the logged model will be added as a new version under that `name`. If no models exist with the given
                 `name`, the given model will be logged as version 1.
-            model_file_or_folder (str): Path to either a single file or a folder containing model files. This folder
-                is usually created using serialization methods of libraries or frameworks e.g. `joblib.dump`,
-                `model.save_pretrained(...)`, `torch.save(...)`, `model.save(...)`
-            framework (Union[enums.ModelFramework, str]): Model Framework. Ex:- pytorch, sklearn, tensorflow etc.
-                The full list of supported frameworks can be found in `truefoundry.ml.enums.ModelFramework`.
-                Can also be `None` when `model` is `None`.
+            model_file_or_folder (Union[str, BlobStorageModelDirectory]):
+                str:
+                    Path to either a single file or a folder containing model files.
+                    This folder is typically created using serialization methods from libraries or frameworks,
+                    e.g., `joblib.dump`, `model.save_pretrained(...)`, `torch.save(...)`, or `model.save(...)`.
+                BlobStorageModelDirectory:
+                    uri (str): URI to the model file or folder in a storage integration associated with the specified ML Repo.
+                        The model files or folder must reside within the same storage integration as the specified ML Repo.
+                        Accepted URI formats include `s3://integration-bucket-name/prefix/path/to/model` or `gs://integration-bucket-name/prefix/path/to/model`.
+                        If the URI points to a model in a different storage integration, an error will be raised indicating "Invalid source URI."
+            framework (Optional[Union[ModelFramework, ModelFrameworkType]]): Framework used for model serialization.
+                Supported frameworks values (ModelFrameworkType) can be imported from `from truefoundry.ml import *`.
+                Supported frameworks can be found in `truefoundry.ml.enums.ModelFramework`.
+                Can also be `None` if the framework is not known or not supported.
+                **Deprecated**: Prefer `ModelFrameworkType` over `enums.ModelFramework`.
             additional_files (Sequence[Tuple[Union[str, Path], Optional[str]]], optional): A list of pairs
                 of (source path, destination path) to add additional files and folders
                 to the model version contents. The first member of the pair should be a file or directory path
@@ -1064,12 +1079,12 @@ class MlFoundryRun:
             run=self,
             name=name,
             model_file_or_folder=model_file_or_folder,
-            framework=framework,
             additional_files=additional_files,
             description=description,
             metadata=metadata,
             step=step,
             progress=progress,
+            framework=framework,
         )
         logger.info(f"Logged model successfully with fqn {model_version.fqn!r}")
         return model_version

truefoundry/ml/model_framework.py

@@ -0,0 +1,169 @@
+import warnings
+from typing import Literal, Optional, Union, get_args
+
+from truefoundry.ml import ModelFramework
+from truefoundry.ml.autogen.entities import artifacts as autogen_artifacts
+from truefoundry.pydantic_v1 import BaseModel, Field
+
+
+class FastAIFramework(autogen_artifacts.FastAIFramework):
+    """FastAI model Framework"""
+
+    type: Literal["fastai"] = "fastai"
+
+
+class GluonFramework(autogen_artifacts.GluonFramework):
+    """Gluon model Framework"""
+
+    type: Literal["gluon"] = "gluon"
+
+
+class H2OFramework(autogen_artifacts.H2OFramework):
+    """H2O model Framework"""
+
+    type: Literal["h2o"] = "h2o"
+
+
+class KerasFramework(autogen_artifacts.KerasFramework):
+    """Keras model Framework"""
+
+    type: Literal["keras"] = "keras"
+
+
+class LightGBMFramework(autogen_artifacts.LightGBMFramework):
+    """LightGBM model Framework"""
+
+    type: Literal["lightgbm"] = "lightgbm"
+
+
+class ONNXFramework(autogen_artifacts.ONNXFramework):
+    """ONNX model Framework"""
+
+    type: Literal["onnx"] = "onnx"
+
+
+class PaddleFramework(autogen_artifacts.PaddleFramework):
+    """Paddle model Framework"""
+
+    type: Literal["paddle"] = "paddle"
+
+
+class PyTorchFramework(autogen_artifacts.PyTorchFramework):
+    """PyTorch model Framework"""
+
+    type: Literal["pytorch"] = "pytorch"
+
+
+class SklearnFramework(autogen_artifacts.SklearnFramework):
+    """Sklearn model Framework"""
+
+    type: Literal["sklearn"] = "sklearn"
+
+
+class SpaCyFramework(autogen_artifacts.SpaCyFramework):
+    """SpaCy model Framework"""
+
+    type: Literal["spacy"] = "spacy"
+
+
+class StatsModelsFramework(autogen_artifacts.StatsModelsFramework):
+    """StatsModels model Framework"""
+
+    type: Literal["statsmodels"] = "statsmodels"
+
+
+class TensorFlowFramework(autogen_artifacts.TensorFlowFramework):
+    """TensorFlow model Framework"""
+
+    type: Literal["tensorflow"] = "tensorflow"
+
+
+class TransformersFramework(autogen_artifacts.TransformersFramework):
+    """Transformers model Framework"""
+
+    type: Literal["transformers"] = "transformers"
+
+
+class XGBoostFramework(autogen_artifacts.XGBoostFramework):
+    """XGBoost model Framework"""
+
+    type: Literal["xgboost"] = "xgboost"
+
+
+# Union of all the model frameworks
+
+
+ModelFrameworkType = Union[
+    FastAIFramework,
+    GluonFramework,
+    H2OFramework,
+    KerasFramework,
+    LightGBMFramework,
+    ONNXFramework,
+    PaddleFramework,
+    PyTorchFramework,
+    SklearnFramework,
+    SpaCyFramework,
+    StatsModelsFramework,
+    TensorFlowFramework,
+    TransformersFramework,
+    XGBoostFramework,
+]
+
+
+class _ModelFramework(BaseModel):
+    __root__: ModelFrameworkType = Field(discriminator="type")
+
+    @classmethod
+    def to_model_framework_type(
+        cls,
+        framework: Optional[Union[str, ModelFramework, "ModelFrameworkType"]] = None,
+    ) -> Optional["ModelFrameworkType"]:
+        """
+        Converts a ModelFramework or string representation to a ModelFrameworkType object.
+
+        Args:
+            framework (Optional[Union[str, ModelFramework, ModelFrameworkType]]): ModelFrameworkType or equivalent input.
+                Supported frameworks can be found in `truefoundry.ml.enums.ModelFramework`.
+                May be `None` if the framework is unknown or unsupported.
+                **Deprecated**: Prefer passing a `ModelFrameworkType` instance.
+
+        Returns:
+            ModelFrameworkType corresponding to the input, or None if the input is None.
+        """
+        if framework is None:
+            return None
+
+        # Issue a deprecation warning for str and ModelFramework types
+        if isinstance(framework, (str, ModelFramework)):
+            warnings.warn(
+                "Passing a string or ModelFramework Enum is deprecated. Please use a ModelFrameworkType object.",
+                DeprecationWarning,
+                stacklevel=2,
+            )
+
+        # Convert string to ModelFramework
+        if isinstance(framework, str):
+            framework = ModelFramework(framework)
+
+        # Convert ModelFramework to ModelFrameworkType
+        if isinstance(framework, ModelFramework):
+            if framework == ModelFramework.UNKNOWN:
+                return None
+            return cls.parse_obj({"type": framework.value}).__root__
+
+        # Directly return if already a ModelFrameworkType
+        if isinstance(framework, get_args(ModelFrameworkType)):
+            return framework
+
+        raise ValueError(
+            "framework must be a string, ModelFramework enum, or ModelFrameworkType object"
+        )
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> ModelFrameworkType:
+        """Create an instance of ModelFramework from a dict"""
+        if obj is None:
+            return None
+
+        return cls.parse_obj(obj).__root__

{truefoundry-0.4.6.dist-info → truefoundry-0.5.0rc2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.4.6
+Version: 0.5.0rc2
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com

{truefoundry-0.4.6.dist-info → truefoundry-0.5.0rc2.dist-info}/RECORD

@@ -123,26 +123,26 @@ truefoundry/langchain/truefoundry_embeddings.py,sha256=8nRaZ7W1ao1WF0LHk6nNel1Lu
 truefoundry/langchain/truefoundry_llm.py,sha256=CJXyCgXIMbDsVRuuvEA5PKJsf6aRyVlYuG7zC4qtZXE,3802
 truefoundry/langchain/utils.py,sha256=PGLDe9chZ3BuUjakexOGpIqZRFoHEgu-zJ9yKdpLLmM,1329
 truefoundry/logger.py,sha256=u-YCNjg5HBwE70uQcpjIG64Ghos-K2ulTWaxC03BSj4,714
-truefoundry/ml/__init__.py,sha256=2A1l7pgqbVRt3cRW_0Lxg92hyJEkMxkCUh1EFprrmc0,942
+truefoundry/ml/__init__.py,sha256=ZMT9C7q9YnwfnfruI0xipizSo1OO__f8Ao-oE65DZyg,1836
 truefoundry/ml/artifact/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/ml/artifact/truefoundry_artifact_repo.py,sha256=VU8i3jnY62MLfzA3rxXuUjdqLz8Yaw4zqqPWSsf0mBg,45850
 truefoundry/ml/autogen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/ml/autogen/client/__init__.py,sha256=B6Z0oUmE-S9YqUCDRpKDuTyNWX2Fmx4es7nWx7Bhy10,17600
-truefoundry/ml/autogen/client/api/__init__.py,sha256=3sMSljMIS3UHYeF0BcNvrPPx6VbBSRt_1IfDn-13Kyc,752
+truefoundry/ml/autogen/client/__init__.py,sha256=0-I7KuPudFU1I1Y4qPbkvAK54E7wzjwIdjnVB57fWGA,18958
+truefoundry/ml/autogen/client/api/__init__.py,sha256=NyMBxBmIzB1o5LzZZwz9LiydHB3-hPqc_sevsnY6Jrw,746
 truefoundry/ml/autogen/client/api/auth_api.py,sha256=zpWzJhUmW6HHMY_atlUf0B25k77E1kue2hmix5I5Ih0,7017
-truefoundry/ml/autogen/client/api/deprecated_api.py,sha256=JCQ39Y3VHdgJ1zM4XVabdTl6QpOHtQFjh04XUb6mN9A,24218
+truefoundry/ml/autogen/client/api/deprecated_api.py,sha256=0qooHNzm5mgweKUEDjnESHPeUqrlzuNEsDJwW9Cm9Vs,24232
 truefoundry/ml/autogen/client/api/experiments_api.py,sha256=mRKS8qGzcFJUpTWjfQoFYm2KI8jTHxBzA0l_foSCLM0,74880
+truefoundry/ml/autogen/client/api/generate_code_snippet_api.py,sha256=C8iABP-pYIyIoTlL8PFi3ZBG_6B_dkQf09eErNeoj7A,22316
 truefoundry/ml/autogen/client/api/health_api.py,sha256=IAPhRAo9CLUT5ipVR1fCf-qDx57UR0wg5ekhtUl8lug,11554
 truefoundry/ml/autogen/client/api/metrics_api.py,sha256=q3L38eD-2hu4_9YvcSdnWDYXD2V8il-X9reinOAABek,14908
 truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py,sha256=smLST3wBZnMw0VrTDR5y1IuJvzV6n89nLWqUMB5mavY,313547
-truefoundry/ml/autogen/client/api/python_deployment_config_api.py,sha256=8P53aegaaZp1LEV9_GDNSCofgxRXP3Atyprc2bUvvpI,8348
 truefoundry/ml/autogen/client/api/run_artifacts_api.py,sha256=x-vVnY2LEFChZxiiFauswRWwFz6Qqh30PKXjzuTvxmc,8799
 truefoundry/ml/autogen/client/api/runs_api.py,sha256=-aghrZ2VYuZOw_vBtOzWDsnK7Ji29oZQxK2CLRgyo2w,119232
 truefoundry/ml/autogen/client/api_client.py,sha256=8qg-WpadDuKgbRt5yABJ4wVS4IRxd5e7rbwe8zBW2wA,28851
 truefoundry/ml/autogen/client/api_response.py,sha256=KRyvecPMXF05PaxILHZ8JHoP4rgKBjKONMgG83aU-rM,844
 truefoundry/ml/autogen/client/configuration.py,sha256=V1oaEnxt-NfpaNmp-EZpf2glovzVhM2coWYt8HBNB4M,15723
 truefoundry/ml/autogen/client/exceptions.py,sha256=XbCbDHhYT3BVejdoGNPgEa4oS56ypkwFdxk1iOc_tFY,5355
-truefoundry/ml/autogen/client/models/__init__.py,sha256=PNukskNa-cv5uP25VmJXrZAZygCoyPis4LkMVLRA7M4,16216
+truefoundry/ml/autogen/client/models/__init__.py,sha256=ZNjoIgoxY6uxmChlRP7q5ea1g2EZvNARDvHKIYujGXU,17580
 truefoundry/ml/autogen/client/models/add_custom_metrics_to_model_version_request_dto.py,sha256=_ISDspicTGjBCYYXubKfRYYSSQVyW3AvG-jFh47-Zfc,2163
 truefoundry/ml/autogen/client/models/add_features_to_model_version_request_dto.py,sha256=rU0h96pEE8K1Ukw2pzDSjq0e6BgtDEuOctI-aZMrpUY,2653
 truefoundry/ml/autogen/client/models/agent.py,sha256=fnMWdEPe5Iw50WKydtu7QAxU419j3ju1IukpChUnGqY,3871
@@ -165,6 +165,7 @@ truefoundry/ml/autogen/client/models/body_get_search_runs_get.py,sha256=aGu5Ah2Q
 truefoundry/ml/autogen/client/models/chat_prompt.py,sha256=PFpvfDGWLSUWYTHEFAOtScJeTYjUUziQD_Meii2D0gw,5974
 truefoundry/ml/autogen/client/models/chat_prompt_messages_inner.py,sha256=nk4hWTl1i3IfFbXVE6cSC-z-QxW4m_YAF05wXdUYYCc,5751
 truefoundry/ml/autogen/client/models/columns_dto.py,sha256=bjSjp-DA2YdjohahziETrfMkKNWFAjDL7XQgSUhvNJA,2070
+truefoundry/ml/autogen/client/models/command.py,sha256=q6yFdBsXZkMFCK8DfNx1JmW8WIHY6scvfQL9NUmQWik,4790
 truefoundry/ml/autogen/client/models/content.py,sha256=vns4BcEnkkZissELyrvak7ZYPyn-lrRcdAGmdYsIBoU,4945
 truefoundry/ml/autogen/client/models/content1.py,sha256=Sh22V-dSQr8zz2N2BgqeTtauE_9dijwydAjpW6uH1z0,4951
 truefoundry/ml/autogen/client/models/content2.py,sha256=T55JENSS5E9HRkefccw24L7PB-rNp3VP1ppXfV7BNHk,5920
@@ -184,6 +185,7 @@ truefoundry/ml/autogen/client/models/create_python_deployment_config_request_dto
 truefoundry/ml/autogen/client/models/create_python_deployment_config_response_dto.py,sha256=5fnCJ7ybEcRA03s8ZLGdt1w8_Xz5nt6-PPmvA6Yxt60,2051
 truefoundry/ml/autogen/client/models/create_run_request_dto.py,sha256=OqhHezsPGygl5m3BlEHmKGlCNf2t5FUFQJLwdmHfggg,2841
 truefoundry/ml/autogen/client/models/create_run_response_dto.py,sha256=cFzIkhbiB_IEmQSx1asNAzPMN-MzCBtqWj2PJ3oTLTM,2169
+truefoundry/ml/autogen/client/models/create_workflow_task_config_request_dto.py,sha256=u-jNJgSeUUCbvyXq9Tg0TUKYh1dTRsrLktXWIkvOoWA,2193
 truefoundry/ml/autogen/client/models/dataset_dto.py,sha256=V3dIdK93YQ8mtolY8Onz2N17HQYm3bLUMn98dVg7tGo,4100
 truefoundry/ml/autogen/client/models/dataset_response_dto.py,sha256=5h3eHpiG4Zzqn5Hm0WpZgyXGjxujKem42_8YnGq5KLo,2205
 truefoundry/ml/autogen/client/models/delete_artifact_versions_request_dto.py,sha256=JF2dQuEe_W1c2ox9zZDER_pZSTbKwSnns8hIfG5o7oA,1905
@@ -196,12 +198,13 @@ truefoundry/ml/autogen/client/models/experiment_dto.py,sha256=bl4-Hp1EHl8CEKgWq6
 truefoundry/ml/autogen/client/models/experiment_id_request_dto.py,sha256=ZMH827n_UTpDI30UnkuOam-4ANBKCDgocIzI8StxFR8,1891
 truefoundry/ml/autogen/client/models/experiment_response_dto.py,sha256=wuflV6_f8PQq061-wU2GzNY4BZi8SG8ARCIbSQN1oT4,2268
 truefoundry/ml/autogen/client/models/experiment_tag_dto.py,sha256=nEpCkeZ9ficIDkjmmLfkJeNNokd-Rhgr-cepPWG6L3M,1902
-truefoundry/ml/autogen/client/models/external_model_source.py,sha256=Vzn1sbjMtQYuMEVjd9hxH0TupGG7vwzqadwE4l9vzs8,2286
+truefoundry/ml/autogen/client/models/external_model_source.py,sha256=WE3JoHmaCDIeoHpcSOtkDsNocBoaTMzqiIuAiq6qIro,2336
+truefoundry/ml/autogen/client/models/fast_ai_framework.py,sha256=pl--nyTJF687SNmX0v1ICISmUCVc6ekF8VH_Rl1kHEQ,2170
 truefoundry/ml/autogen/client/models/feature_dto.py,sha256=XM7fF71P_bYP3UlO1ILjnD3DJNpf2tGwAWbrkQyvyA4,1924
 truefoundry/ml/autogen/client/models/feature_value_type.py,sha256=3ZKxxEfrrWk8ePlV7n11wa-FdDH0SRLyHRqSvIU7F-M,732
 truefoundry/ml/autogen/client/models/file_info_dto.py,sha256=7oc8venicsFVk8zT9wHNhHnZGtFkFlqimFnS7ozGL9k,2156
 truefoundry/ml/autogen/client/models/finalize_artifact_version_request_dto.py,sha256=tArvzRPp80PpfJ4CoMwwaKqPAfzonq-8C-yKBEjs9co,3877
-truefoundry/ml/autogen/client/models/framework.py,sha256=Z_91ZrPWB22mHnKWBgbXHhY5NzHdmfTATBd2iQPNIaU,5131
+truefoundry/ml/autogen/client/models/framework.py,sha256=W_KrqH1qKRvlrssCCWFIlmPaeAw0eiM6VqS3pbdlMXU,14720
 truefoundry/ml/autogen/client/models/get_experiment_response_dto.py,sha256=FhJpoUeRbZV1s1KR3Y07Kv6U4enLbbgdpGbguT9us0A,2923
 truefoundry/ml/autogen/client/models/get_latest_run_log_response_dto.py,sha256=g5NG252VzjGHiA-w5nhxmZxGrowJO9XCHDQTjf8y2KE,2266
 truefoundry/ml/autogen/client/models/get_metric_history_response.py,sha256=rTchO8o4sjGsNaZj8TW-VGtGGxsOBjUGzgaAWNtBZ-k,2428
@@ -214,12 +217,16 @@ truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_read_request_dt
 truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_read_response_dto.py,sha256=1EQ33wOq7VpAlt7o-5XCBZ2mpn5GXKrOUfULbp4gA2g,2627
 truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_write_response_dto.py,sha256=tRud_0Qrr4DtXD1SMSnouVlxrV6WaL5FG2mG626CFZ0,2635
 truefoundry/ml/autogen/client/models/get_tenant_id_response_dto.py,sha256=zEG9rogifl100cOqshYDNJE-faxGF3ayRb1RNWCtaXo,2142
+truefoundry/ml/autogen/client/models/gluon_framework.py,sha256=YIc4x1A7kW4F9Bgltr6mU0UgL2sxJMSd8qyZPPtq900,2149
+truefoundry/ml/autogen/client/models/h2_o_framework.py,sha256=DrJgVIP62OgyE0YdV22gD1K4u6JHVfRITunI5moj5mk,2125
 truefoundry/ml/autogen/client/models/http_validation_error.py,sha256=bU-uUlz_qeFba3iHdBT7cHjLu6WZ-JmImYqgZ-kCvxM,2466
 truefoundry/ml/autogen/client/models/image_content_part.py,sha256=_DXwAxeuagwRWMkDxaz--i_OitlyNkPd3puewVV_fwM,2716
 truefoundry/ml/autogen/client/models/image_url.py,sha256=TN8iEdBUmtKdTFlApUodaC4o6X_Ggd3OSOKVPv7laJE,2170
 truefoundry/ml/autogen/client/models/internal_metadata.py,sha256=6OizrIzA3QxlEnwNFXCZGOMJuUDSYHctl_30q27uBDI,6232
+truefoundry/ml/autogen/client/models/keras_framework.py,sha256=7TyZHDE3A__fF3lCX-scmCCfLgHpFwBhr0YSLNCoDBA,2149
 truefoundry/ml/autogen/client/models/latest_run_log_dto.py,sha256=wE8T8bANTb9u14Jv7DNyKWekZiUAvrzvTcE_H1cRhn4,2326
 truefoundry/ml/autogen/client/models/library_name.py,sha256=JKmQMgEj6RcmSv68THAcs1gmh3P3CNrgJ_Rh5VWu9Fs,823
+truefoundry/ml/autogen/client/models/light_gbm_framework.py,sha256=PVW9r0FnYxvk5LRRw5ioOflMJZ1ZR19K1uVmdEHqe38,2194
 truefoundry/ml/autogen/client/models/list_artifact_versions_request_dto.py,sha256=nODOEA-tT8XEJKgUXXjvlzT5ecEwz1b9zu-l3RgQl1o,3622
 truefoundry/ml/autogen/client/models/list_artifact_versions_response_dto.py,sha256=G560h_rGn6-X-cxpuddGU8Fp2eBKXvC3RWcJZdZQJBE,2940
 truefoundry/ml/autogen/client/models/list_artifacts_request_dto.py,sha256=o3Jz3VsAa3aJB3YZ6NI6t8TLLoMGeg64jYYZibeo3Oc,2982
@@ -253,16 +260,19 @@ truefoundry/ml/autogen/client/models/model_dto.py,sha256=TUO74MDqe8XCVJBB2O7Tiqi
 truefoundry/ml/autogen/client/models/model_response_dto.py,sha256=osrTxfygkuhxWj6SkRBALrSnFVPH4LSK6qTufTeZuJg,2163
 truefoundry/ml/autogen/client/models/model_schema_dto.py,sha256=ElEPK7fwuf3eiohNBBHJyl-nhNqzGnDLzZVzuQl-xC0,2620
 truefoundry/ml/autogen/client/models/model_version_dto.py,sha256=6kI1Z2czYAieJZBaVvcV8Y2eeNEUDyjPuTLHTdyWnOE,7282
-truefoundry/ml/autogen/client/models/model_version_manifest.py,sha256=yJN2C_ETddLAeSR4wDuzV3--F2MyiN-XzIJlNayG6Ok,3945
+truefoundry/ml/autogen/client/models/model_version_manifest.py,sha256=Jd8oJj9-awzg-jP_hhLEMH3PSXir2p5Kc5o0YEB9ZRA,3957
 truefoundry/ml/autogen/client/models/model_version_response_dto.py,sha256=D6XOCyggxqTkbePuypqYSHYh1PYeDv7R_J32q61sDM0,2320
 truefoundry/ml/autogen/client/models/multi_part_upload_dto.py,sha256=Ckq405vud8RQmMyKCJQJBlW5iXO7Y2mlgo8eVkiMvxg,3738
 truefoundry/ml/autogen/client/models/multi_part_upload_response_dto.py,sha256=VjH0kvl7rMjgDHjYGHnsh7KsZ5-qn-k3ksdGLJ49nIM,2431
 truefoundry/ml/autogen/client/models/multi_part_upload_storage_provider.py,sha256=BA-a-LUi8Q8wTmrExiidRWmCQUAmAqn5qOTwgQM1K6U,796
 truefoundry/ml/autogen/client/models/notify_artifact_version_failure_dto.py,sha256=M5PaBiH0KYigTpmg6Rm6n2nM6I9x6pLjI-oNqY4006o,1897
+truefoundry/ml/autogen/client/models/onnx_framework.py,sha256=-z-S50V11nploHrPCoY0Mq4KQCrM8tY5R_rEc1p5v0g,2137
 truefoundry/ml/autogen/client/models/openapi_spec.py,sha256=jWD8fO4Rjx7NXg-kvJ1KEDT7ysHZcYbQ-EnEk3MxMqs,4961
+truefoundry/ml/autogen/client/models/paddle_framework.py,sha256=cTTyFLaXHjSo8Kc-zU_fDBF4nAcXu74CV0AlV4JwYkc,2176
 truefoundry/ml/autogen/client/models/param_dto.py,sha256=A8ysdVT16LZDB78Cd18ot2wf6emWNGKPv70IQ6wcEak,1787
 truefoundry/ml/autogen/client/models/parameters.py,sha256=u3DF-XvyZ9w1GgEJ2WCQyYX6RXbkBw7B2yWKTFjVfP4,2736
 truefoundry/ml/autogen/client/models/prediction_type.py,sha256=ayeIx_mdlOYXV82yoHoTQEYS9mIzRcTuYB9V0wIe_GE,722
+truefoundry/ml/autogen/client/models/py_torch_framework.py,sha256=2sctpFx43y2C_fOh246M7KKHGlujz41KPJJWnOL4hJ0,2182
 truefoundry/ml/autogen/client/models/resolve_agent_app_response_dto.py,sha256=sggWYqC5R_BtH99xne3G_lgZYuyJjT2wozwDtvfB_Vs,2360
 truefoundry/ml/autogen/client/models/restore_run_request_dto.py,sha256=7iBbil4YfaOHJwP0ifPHp-D1Kt-ZiSRjAaVuU7_idhw,1825
 truefoundry/ml/autogen/client/models/run_data_dto.py,sha256=iPFm5f4x2ymRqea6EDZlcjWYN_-f3t9JZckwskXSTKY,3533
@@ -277,31 +287,36 @@ truefoundry/ml/autogen/client/models/search_runs_response_dto.py,sha256=u0l11v7p
 truefoundry/ml/autogen/client/models/set_experiment_tag_request_dto.py,sha256=nrmi_NxLD1fI2gwlpdqFSMnBS11gRkjS4_GQFHgBcXs,2118
 truefoundry/ml/autogen/client/models/set_tag_request_dto.py,sha256=IRgAdMcWBxmjNV6nZJej4pcNfLmZwrelEZ3otwt7eeE,2144
 truefoundry/ml/autogen/client/models/signed_url_dto.py,sha256=9oHoXBj07xTdc04rqOqJO3eOjQWXCyWPhfHg-6qX60w,1897
-truefoundry/ml/autogen/client/models/source.py,sha256=OqcgjSiQwji_jpfcL29HKlAebURPChuQPqmQzd7CSPk,5961
+truefoundry/ml/autogen/client/models/sklearn_framework.py,sha256=8-GvfwN5fHVDiOXPvVa9rBeG0VEavwwTsfEp3Nl0uO8,2187
+truefoundry/ml/autogen/client/models/source.py,sha256=8vQ6Fcky8kYAQiATuphxuhxKGRhH4WmcTiUa9N8iZQM,5095
+truefoundry/ml/autogen/client/models/spa_cy_framework.py,sha256=JM1DLF8JL4c7-JZN-dQX_7NF8jIS6RP-mEk1uoWQpaU,2149
+truefoundry/ml/autogen/client/models/stats_models_framework.py,sha256=INndD-nZFAfRbKZXss4erx4xtotb_hlRFlEvitKrIzQ,2230
 truefoundry/ml/autogen/client/models/stop.py,sha256=UlAjRWI6Gjc92D2UME6qRAIfd6GoDI3VBWYkJqgixmU,4760
 truefoundry/ml/autogen/client/models/store_run_logs_request_dto.py,sha256=Cfm95a9T4fDiARBJ9vd3n9Pv7QI69TSVKWi0AG_cmvs,2591
 truefoundry/ml/autogen/client/models/subject.py,sha256=gG1pRl6lPEI1_td2sb752vJeKMKgLWDaoFwepnS4zIQ,2415
 truefoundry/ml/autogen/client/models/subject_type.py,sha256=TOhgDmWSy67NjBCTSbwtF2-cBBB2-FoLJVFDVpJRrsQ,710
 truefoundry/ml/autogen/client/models/system_message.py,sha256=yxRCJPl36FGv2zrsuHbLown88wmnrIpWlKEVRf9WgFQ,2801
-truefoundry/ml/autogen/client/models/tensorflow_framework.py,sha256=8F0CHJj_J0V57GePBuMjSSLmCRuwIot10ghywlBFIeQ,2251
+truefoundry/ml/autogen/client/models/tensor_flow_framework.py,sha256=lvABZqmVED7Hg7aDZ7bovN6usG_TZCWI4B8Ew6NAymw,2218
 truefoundry/ml/autogen/client/models/text.py,sha256=HAIK-w_TxB1kXGvVNK1d2cweoQi-GJ9d-0QSZl21u68,4862
 truefoundry/ml/autogen/client/models/text_content_part.py,sha256=fRQu22tiV3rDKmTgVZSLHX-cMcha3q53W8qWNCk5wmU,2492
-truefoundry/ml/autogen/client/models/transformers_framework.py,sha256=5n4h3IvcCRr4Dca9Sv1EGUp07rEgv3c41U47NmGTsu0,2916
-truefoundry/ml/autogen/client/models/truefoundry_model_source.py,sha256=kKxDCkLKTLho2SkVO0df6DoD8czwKbaacbjNRqBMhgY,2396
+truefoundry/ml/autogen/client/models/transformers_framework.py,sha256=ialS6K3F64iLSe19qVXeuXNctEq6IzMFi7cBhP00bdg,2903
+truefoundry/ml/autogen/client/models/trigger_job_run_config_request_dto.py,sha256=1HtO6zfW6Ulq8Gm_UL0YqVrse0795vZkMxgCQ0GKD9w,2864
+truefoundry/ml/autogen/client/models/trigger_job_run_config_response_dto.py,sha256=P35xzE8MyS_HtxCcXC9fZORPX2wbf93LbCZCRggkQxw,2144
+truefoundry/ml/autogen/client/models/truefoundry_model_source.py,sha256=Jc3sP2n05kQz92nHaU5RCcwP9bfeIBFUuxfvwdojub8,2418
 truefoundry/ml/autogen/client/models/update_artifact_version_request_dto.py,sha256=AhXj2afmrKskBi5uMT0-mSywZyz3g17lFE3uMRQ5pEM,2246
 truefoundry/ml/autogen/client/models/update_dataset_request_dto.py,sha256=MF-rjJGUP6sDZVH3xP4Q8AEC9CV2PXEO9cXrR6yK57A,2182
 truefoundry/ml/autogen/client/models/update_experiment_request_dto.py,sha256=QE_kZzTDdiXKtQ2U-zbI_gb5vxS5Yv9YupYvD22RkFs,2198
 truefoundry/ml/autogen/client/models/update_model_version_request_dto.py,sha256=0xWqHHflnpO8FyT79psPiPBIGW6Aiovs-b6H7sjMSDQ,3589
 truefoundry/ml/autogen/client/models/update_run_request_dto.py,sha256=fn4VB4Zlllr6Bt_l3aYkkHxAkU9w7vtcqu4iw8BvQWY,2318
 truefoundry/ml/autogen/client/models/update_run_response_dto.py,sha256=wolIFiihGZCBBkuj7mJEYCmGAOqDLFo_bNLcH4LeXSo,2231
-truefoundry/ml/autogen/client/models/upload_model_source.py,sha256=zn2AigQc8uhJiGXdE57X-K4pLPrIK9TuD8g7zNt09-U,2103
 truefoundry/ml/autogen/client/models/url.py,sha256=zMyOmdVkp1ANnQnc9GrHt42xlVwES7FTjrpOp_OtHqo,4953
 truefoundry/ml/autogen/client/models/user_message.py,sha256=6QOmKT_SDfAVZA8V3Bzofm7_c-Qznn4KsE12NmHG4FM,2783
 truefoundry/ml/autogen/client/models/validation_error.py,sha256=mFjwoc8g2-Usu1HXZhOQKQ4TGvLy4lwCzk8dHrJ69aA,2597
 truefoundry/ml/autogen/client/models/validation_error_loc_inner.py,sha256=nThJ5Gmy8W2Wok-ZOI4sK7uRe1BAkLS0qzq-XZbq8zs,4915
+truefoundry/ml/autogen/client/models/xg_boost_framework.py,sha256=kXijWcmEd5mc6Rks0mYQLqgYp1v7bxp31HQAmaJLZ3s,2182
 truefoundry/ml/autogen/client/rest.py,sha256=9goba8qHjQuVx5O_yRaTKu7PvBnb7r7swfy3dwuTEgk,14281
-truefoundry/ml/autogen/client_README.md,sha256=ehPzJCsvWR_ghMgIcP-UXMG4AgqcJI6cbpuz1rcsqME,33871
-truefoundry/ml/autogen/entities/artifacts.py,sha256=yXLrEsX28aCxpgRA8x8YM5M55LLlLiWoJqEWRAZu6_Y,16147
+truefoundry/ml/autogen/client_README.md,sha256=h8EUUiMl1vHri3Jrt0PjkFD-CoXAf5WXSgG4iRqlouc,35564
+truefoundry/ml/autogen/entities/artifacts.py,sha256=qc4XW3X8BR-zE32tNbCBMJ5kc_aPehDOBhyugvAYY-8,16646
 truefoundry/ml/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/ml/cli/cli.py,sha256=ckBcjUpqfhgrPE1okqT_G2iouOLt-0KjpLhHp2YdVFU,256
 truefoundry/ml/cli/commands/__init__.py,sha256=diDUiRUX4l6TtNLI4iF-ZblczkELM7FRViJ-8gGNJQY,82
@@ -321,7 +336,7 @@ truefoundry/ml/log_types/artifacts/artifact.py,sha256=n2EwAupOigDcUXQQFIIESRqp7z
 truefoundry/ml/log_types/artifacts/constants.py,sha256=qKxQ5mMvJE4j83BvGW3qNTKunxCiBg_EEjTdgbgJtyE,1036
 truefoundry/ml/log_types/artifacts/dataset.py,sha256=a4dxd2EN8p7Ci-cLGGiDOboN3t0395_XhWE1dmTw1Q4,13112
 truefoundry/ml/log_types/artifacts/general_artifact.py,sha256=_EOlNGMg2PFjlevlD6LEOsaQBBwRWs60PkzJases6tE,3927
-truefoundry/ml/log_types/artifacts/model.py,sha256=Q4Phy8-5E8hDxHC-Zg-V1uj_-V3p1z6Wh6JC-zot0Vs,21912
+truefoundry/ml/log_types/artifacts/model.py,sha256=fcMc63BKqcbjUygpA6w0SFuoWZDvyxJvlPHyh7bNKv4,21161
 truefoundry/ml/log_types/artifacts/model_extras.py,sha256=TIE73bLKfwIVzNiVcjmaZ841A70BHBwu4XAM6ZAQRFI,1045
 truefoundry/ml/log_types/artifacts/utils.py,sha256=3FVOYlfcvZiW8vTbMe7Ft1cjNPR2GW7J69M3dyAY-qc,6299
 truefoundry/ml/log_types/image/__init__.py,sha256=fcOq8yQnNj1rkLcPeIjLXBpdA1WIeiPsXOlAAvMxx7M,76
@@ -333,8 +348,9 @@ truefoundry/ml/log_types/plot.py,sha256=HuYvvRA5r8V0xAIuuqMME2IHb9d3SfGHUiuEkOP3
 truefoundry/ml/log_types/pydantic_base.py,sha256=eBlw_AEyAz4iJKDP4zgJOCFWcldwQqpf7FADW1jzIQY,272
 truefoundry/ml/log_types/utils.py,sha256=xjJ21jdPScvFmw3TbVh5NCzbzJwaqiXJyiiT4xxX1EI,335
 truefoundry/ml/logger.py,sha256=VT-BF3BnBYTWVq87O58F0c8uXMu94gYzsiFlGY3_7Ao,458
-truefoundry/ml/mlfoundry_api.py,sha256=lVsD7WBqAZMNSEauKeiF1jCg_Fnt6IFEPRr4mwn1IMU,60820
-truefoundry/ml/mlfoundry_run.py,sha256=rNJl130iJkpjW3MNoe5-d_J9VJJQBqWHEJCfYeiZCbE,45123
+truefoundry/ml/mlfoundry_api.py,sha256=eLzbXPSaUWBuIZQKld0r30StOHD-dXiS9zQL_Wq4Zk4,62037
+truefoundry/ml/mlfoundry_run.py,sha256=FDTEBflyKRA7nP8qCm834NyYjpyy9540J_WnagB25oI,46195
+truefoundry/ml/model_framework.py,sha256=YCdRD31rQbPQK-ehNLf1YHuOcPgserFjdPgrrB2Twqg,4737
 truefoundry/ml/run_utils.py,sha256=0W208wSLUrbdfk2pjNcZlkUi9bNxG2JORqoe-5rVqHI,2423
 truefoundry/ml/session.py,sha256=F83GTC5WwGBjnJ69Ct8MqMnlutYc56JCc6YhEY1Wl-A,5394
 truefoundry/ml/validation_utils.py,sha256=XBSUd9OoyriWJpT3M5LKz17iWY3yVMr3hM5vdaVjtf0,12082
@@ -355,7 +371,7 @@ truefoundry/workflow/remote_filesystem/tfy_signed_url_client.py,sha256=ln3zx72jM
 truefoundry/workflow/remote_filesystem/tfy_signed_url_fs.py,sha256=Hf6Dk6Fu6P7DqsK5ULgraf9DStjgigf-kjaRAMBW-RU,8680
 truefoundry/workflow/task.py,sha256=ToitYiKcNzFCtOVQwz1W8sRjbR97eVS7vQBdbgUQtKg,1779
 truefoundry/workflow/workflow.py,sha256=WaTqUjhwfAXDWu4E5ehuwAxrCbDJkoAf1oWmR2E9Qy0,4575
-truefoundry-0.4.6.dist-info/METADATA,sha256=xMHeMDZtsbcVayGsYdRaNH6uMXROl-lx1759kIXYwS8,3098
-truefoundry-0.4.6.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-truefoundry-0.4.6.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.4.6.dist-info/RECORD,,
+truefoundry-0.5.0rc2.dist-info/METADATA,sha256=zrEMchTZaejE_hXfAyMQfMhi7gjmlb0SK9rOY5AtfVA,3101
+truefoundry-0.5.0rc2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+truefoundry-0.5.0rc2.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.5.0rc2.dist-info/RECORD,,