truefoundry 0.5.0rc4__py3-none-any.whl → 0.5.0rc5__py3-none-any.whl

truefoundry/ml/autogen/client/models/model_version_manifest.py

@@ -49,7 +49,19 @@ class ModelVersionManifest(BaseModel):
     step: Optional[conint(strict=True, ge=0)] = Field(
         default=0, description="+label=Step"
     )
-    __properties = ["description", "metadata", "type", "source", "framework", "step"]
+    model_schema: Optional[Dict[str, Any]] = Field(
+        default=None,
+        description="+label=Model Schema +usage=Schema of the model +uiType=Hidden",
+    )
+    __properties = [
+        "description",
+        "metadata",
+        "type",
+        "source",
+        "framework",
+        "step",
+        "model_schema",
+    ]
 
     @validator("type")
     def type_validate_enum(cls, value):
@@ -114,6 +126,7 @@ class ModelVersionManifest(BaseModel):
                 if obj.get("framework") is not None
                 else None,
                 "step": obj.get("step") if obj.get("step") is not None else 0,
+                "model_schema": obj.get("model_schema"),
             }
         )
         return _obj

truefoundry/ml/autogen/entities/artifacts.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  artifacts.json
-#   timestamp: 2024-11-15T09:16:35+00:00
+#   timestamp: 2024-11-22T19:31:48+00:00
 
 from __future__ import annotations
 
@@ -534,6 +534,10 @@ class ModelVersion(BaseArtifactVersion):
         None, description="+label=Framework\n+usage=Framework for the model version"
     )
     step: conint(ge=0) = Field(0, description="+label=Step")
+    model_schema: Optional[Dict[str, Any]] = Field(
+        None,
+        description="+label=Model Schema\n+usage=Schema of the model\n+uiType=Hidden",
+    )
 
 
 class AgentApp(BaseModel):

truefoundry/ml/log_types/artifacts/model.py

@@ -96,6 +96,7 @@ class ModelVersion:
         self._deleted = False
         self._description: str = ""
         self._metadata: Dict[str, Any] = {}
+        self._model_schema: Optional[Dict[str, Any]] = None
         self._framework: Optional[ModelFrameworkType] = None
         self._set_mutable_attrs()
 
@@ -138,6 +139,9 @@ class ModelVersion:
         if self._model_version.manifest:
             self._description = self._model_version.manifest.description or ""
             self._metadata = copy.deepcopy(self._model_version.manifest.metadata)
+            self._model_schema = copy.deepcopy(
+                self._model_version.manifest.model_schema
+            )
             if self._model_version.manifest.framework:
                 self._framework = copy.deepcopy(
                     self._model_version.manifest.framework.actual_instance
@@ -150,6 +154,7 @@ class ModelVersion:
             self._framework = _ModelFramework.to_model_framework_type(
                 self._model_version.model_framework
             )
+            self._model_schema = None
 
     def _refetch_model_version(self, reset_mutable_attrs: bool = True):
         _model_version = self._mlfoundry_artifacts_api.get_model_version_get(
@@ -219,6 +224,23 @@ class ModelVersion:
         _validate_artifact_metadata(value)
         self._metadata = copy.deepcopy(value)
 
+    @property
+    def model_schema(self) -> Optional[Dict[str, Any]]:
+        """Get model_schema for the current model"""
+        return self._model_schema
+
+    @model_schema.setter
+    def model_schema(self, value: Optional[Dict[str, Any]]):
+        """set the model_schema for current model"""
+        if not self._model_version.manifest:
+            warnings.warn(
+                message="This model version was created using an older serialization format. model_schema will not be updated",
+                category=DeprecationWarning,
+                stacklevel=2,
+            )
+            return
+        self._model_schema = copy.deepcopy(value)
+
     @property
     def framework(self) -> Optional["ModelFrameworkType"]:
         """Get the framework of the model"""
@@ -416,6 +438,7 @@ class ModelVersion:
         if self._model_version.manifest:
             self._model_version.manifest.description = self.description
             self._model_version.manifest.metadata = self.metadata
+            self._model_version.manifest.model_schema = self.model_schema
             self._model_version.manifest.framework = (
                 Framework.from_dict(self.framework.dict()) if self.framework else None
             )
@@ -449,6 +472,7 @@ def _log_model_version(  # noqa: C901
     step: Optional[int] = 0,
     progress: Optional[bool] = None,
     framework: Optional[Union[str, ModelFramework, "ModelFrameworkType"]] = None,
+    model_schema: Optional[Dict[str, Any]] = None,
 ) -> ModelVersion:
     if (run and mlfoundry_artifacts_api) or (not run and not mlfoundry_artifacts_api):
         raise MlFoundryException(
@@ -562,6 +586,7 @@ def _log_model_version(  # noqa: C901
         source=_source_cls.from_dict(source.dict()),
         framework=Framework.from_dict(_framework.dict()) if _framework else None,
         step=step,
+        model_schema=model_schema,
     )
     artifact_version_response = mlfoundry_artifacts_api.finalize_artifact_version_post(
         finalize_artifact_version_request_dto=FinalizeArtifactVersionRequestDto(

truefoundry/ml/mlfoundry_api.py

@@ -1232,6 +1232,7 @@ class MlFoundry:
         metadata: Optional[Dict[str, Any]] = None,
         progress: Optional[bool] = None,
         framework: Optional[Union[str, ModelFramework, "ModelFrameworkType"]] = None,
+        model_schema: Optional[Dict[str, Any]] = None,
     ) -> ModelVersion:
         """
         Serialize and log a versioned model under the current ml_repo. Each logged model generates a new version
@@ -1391,6 +1392,7 @@ class MlFoundry:
             step=None,
             progress=progress,
             framework=framework,
+            model_schema=model_schema,
         )
         logger.info(f"Logged model successfully with fqn {model_version.fqn!r}")
         return model_version

truefoundry/ml/mlfoundry_run.py

@@ -936,6 +936,7 @@ class MlFoundryRun:
         step: int = 0,
         progress: Optional[bool] = None,
         framework: Optional[Union[str, ModelFramework, "ModelFrameworkType"]] = None,
+        model_schema: Optional[Dict[str, Any]] = None,
     ) -> ModelVersion:
         # TODO (chiragjn): Document mapping of framework to list of valid model save kwargs
         # TODO (chiragjn): Add more examples
@@ -1088,6 +1089,7 @@ class MlFoundryRun:
             step=step,
             progress=progress,
             framework=framework,
+            model_schema=model_schema,
         )
         logger.info(f"Logged model successfully with fqn {model_version.fqn!r}")
         return model_version

{truefoundry-0.5.0rc4.dist-info → truefoundry-0.5.0rc5.dist-info}/METADATA

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

{truefoundry-0.5.0rc4.dist-info → truefoundry-0.5.0rc5.dist-info}/RECORD

@@ -262,7 +262,7 @@ 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=uglhFSCDbcp9beh-zKJ8eLxsmfwnMkS1si8_GnWTPI8,3961
+truefoundry/ml/autogen/client/models/model_version_manifest.py,sha256=jPE3JYVvv8Wmqc4-w4Ff10HlqqxXaK9EQVkIgN7h9j4,4262
 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
@@ -319,7 +319,7 @@ truefoundry/ml/autogen/client/models/validation_error_loc_inner.py,sha256=nThJ5G
 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=nIyCT1OXBMLV_cadu0OIqQCtXmGm3frJnO0gbJ_7b9Q,35790
-truefoundry/ml/autogen/entities/artifacts.py,sha256=KQvJEVqlQGtq3AKRZ4xhwZC05XtC1JDcuh8_CB1tZ78,17167
+truefoundry/ml/autogen/entities/artifacts.py,sha256=EbGDmD0npfa5BSHau67Y8-dSk6qPJJlJX3I3N4kqAgA,17326
 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
@@ -339,7 +339,7 @@ truefoundry/ml/log_types/artifacts/artifact.py,sha256=6X4lO23DwM7yTgUOY4RRoYMrf9
 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=B4XErLr-m6RmQWtxMTu3wlFRFcqSwPYp6J0OL4Ng6L0,3179
-truefoundry/ml/log_types/artifacts/model.py,sha256=giK2NyEDFTscSSA9M_paraFvI83hUt_xOouUo8aC8gs,21477
+truefoundry/ml/log_types/artifacts/model.py,sha256=PZgqRO8NImvU7yfdjldqJ4Z3ow47GMnSL0A1DeE2DYk,22511
 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
@@ -351,8 +351,8 @@ 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=29orLTRTH_w6A_GbTadKHMSIxcN3Ae82qYWP1ogxtk8,62017
-truefoundry/ml/mlfoundry_run.py,sha256=Cj-4PWbuDyleKOHuJKM25Y2Xik6b04hK3R5RTXlf34o,46188
+truefoundry/ml/mlfoundry_api.py,sha256=_LAXCBu1A-xg8fD3qzjQv9YlghEbSgYe6rQB0YgvXGQ,62111
+truefoundry/ml/mlfoundry_run.py,sha256=jLeDz9raUPKjXGH4PPAFel3Zg030iUWB_QP7FeW0DE0,46282
 truefoundry/ml/model_framework.py,sha256=TosB73eEpTRlsh6v1AY_LnnW_4OoaSlzENnBXqkVRl4,4778
 truefoundry/ml/run_utils.py,sha256=0W208wSLUrbdfk2pjNcZlkUi9bNxG2JORqoe-5rVqHI,2423
 truefoundry/ml/session.py,sha256=F83GTC5WwGBjnJ69Ct8MqMnlutYc56JCc6YhEY1Wl-A,5394
@@ -374,7 +374,7 @@ truefoundry/workflow/remote_filesystem/tfy_signed_url_client.py,sha256=5mBCIc-ON
 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.5.0rc4.dist-info/METADATA,sha256=keTSBb2h2m_q1CiniPQt75UTizz5a2ZyEZzLQSlOkH8,3102
-truefoundry-0.5.0rc4.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-truefoundry-0.5.0rc4.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.5.0rc4.dist-info/RECORD,,
+truefoundry-0.5.0rc5.dist-info/METADATA,sha256=WD9-zZ_Rxg1qbBVQZm_GDVlNvWPzuM2HNC78-Rbg-mk,3102
+truefoundry-0.5.0rc5.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+truefoundry-0.5.0rc5.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.5.0rc5.dist-info/RECORD,,