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

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.0rc6.dist-info}/METADATA

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

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

@@ -127,43 +127,43 @@ truefoundry/ml/__init__.py,sha256=tcFeEcIgBLR92pUDiu7zDSeI1vsSshlCzuIgnSbyE8E,18
 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=OD4hsjMELkJNFkpVzcRGX_0tEIfWxL2n_0XYuy8mMoo,19211
+truefoundry/ml/autogen/client/__init__.py,sha256=6OSFtyl9gWo28jJvvZV6qzsyT0_X_1nQeReE1YvwsKQ,18893
 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=0qooHNzm5mgweKUEDjnESHPeUqrlzuNEsDJwW9Cm9Vs,24232
+truefoundry/ml/autogen/client/api/deprecated_api.py,sha256=mu5x_skNcwz8v1Tr6VP72-tVP7pmBV9muyKy_2NH3F0,38824
 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/mlfoundry_artifacts_api.py,sha256=i9uLXUCuQtiQG15bqutEcGyiE9gDeE_2cebN56iBGk0,299272
 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_client.py,sha256=M31IycWorn10FYS8WbO_Whaipr0tVu2pVWFCfbS7XCo,29167
 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=IH4Cx1qYoYds5nkThxh9yMShRYZ5mGwn5fFFyPvnG6Y,17833
+truefoundry/ml/autogen/client/models/__init__.py,sha256=v5LbkCWeeuytmE6B8QFShtQxJIeXvncENaVG5vGTiPE,17515
 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
-truefoundry/ml/autogen/client/models/agent_app.py,sha256=h-0xdTMDE6FPzvksnbhVSXOBrhGlRmPHx5MMqOY3JRI,3665
-truefoundry/ml/autogen/client/models/agent_open_api_tool.py,sha256=QotUMmGqAMpNsRFwVoNJHg9OeobimS-Wc9Eb85VRD24,5023
-truefoundry/ml/autogen/client/models/agent_open_api_tool_with_fqn.py,sha256=8iHIEpGtQJO8QZS66T0FysqKg1VaxJT44hPu_a6zuwE,5121
-truefoundry/ml/autogen/client/models/agent_with_fqn.py,sha256=_LFRmdT0KNe1CRNBfEt9gFbBvGMiwPInbesHSKrLH-g,3976
+truefoundry/ml/autogen/client/models/add_features_to_model_version_request_dto.py,sha256=OT1-98DyWNfAHz_EgD2gMX2XkrGQ4Re945fhoAl8qSE,2099
+truefoundry/ml/autogen/client/models/agent.py,sha256=lUAbs092yo9hZmqzX6atKebDNf1uEwTf5jLvjOdvoeM,3872
+truefoundry/ml/autogen/client/models/agent_app.py,sha256=LK13z4IZLdE63ySz4IICq0Hn_mC02ijqzA_6Bm1mK3w,3666
+truefoundry/ml/autogen/client/models/agent_open_api_tool.py,sha256=pZaKMzo6mB1t9RGI6mbE_NiWr3wzmVwhIeIFOVCR_iY,5024
+truefoundry/ml/autogen/client/models/agent_open_api_tool_with_fqn.py,sha256=QbRhX7bsE8w8okRS2C2q4edkQXaw61AJLt-hHvBJv5w,5122
+truefoundry/ml/autogen/client/models/agent_with_fqn.py,sha256=zYfLximM_vq4lDaw5_DPnYgvkwIHk_VVPp-inC_6_t4,3977
 truefoundry/ml/autogen/client/models/artifact_dto.py,sha256=JWL8_sITJJXbxywcfRj4xqt8jR9pyBmA4TfwNTdX6CE,4243
 truefoundry/ml/autogen/client/models/artifact_response_dto.py,sha256=T_HLjkHnqiO1bnSXiCUOlwDEHGz89mlUNhBq0SJTapo,2226
 truefoundry/ml/autogen/client/models/artifact_type.py,sha256=LBXGYUoesmL5J-0xzCR8jBST3bjb6HY86RmqSGvytRs,828
 truefoundry/ml/autogen/client/models/artifact_version_dto.py,sha256=GXjjKycpsEgJ60V2hAfL4NMxM2gZKLP2qV2yBb9WgdM,6174
-truefoundry/ml/autogen/client/models/artifact_version_manifest.py,sha256=xqznaihmb9voR_-GqQ3vlDG5ZxvYhjjWCP9nR-WmIIE,3531
+truefoundry/ml/autogen/client/models/artifact_version_manifest.py,sha256=EjmUxCLoN8wt-icGXAqd_q6nPwl3yBDwVqKu7kTYiLA,3532
 truefoundry/ml/autogen/client/models/artifact_version_response_dto.py,sha256=JmiSYdPYG-ki6LCBeF7NtihdQ2LPiHld8cSPivY2GmQ,2421
 truefoundry/ml/autogen/client/models/artifact_version_serialization_format.py,sha256=2Qfgq1nykn0_mKNEJzOxhQwHm_95WIr9wJcHLeFWXQM,774
 truefoundry/ml/autogen/client/models/artifact_version_status.py,sha256=iIcomqhuaJkZH2rT3e6IjQutM79t9hEshW-RCIMKHEY,780
-truefoundry/ml/autogen/client/models/assistant_message.py,sha256=rmCQ4OAbVpjbX3rXiNtv2ePxV-7kq6Eg1Bs_tm_rmNE,2845
+truefoundry/ml/autogen/client/models/assistant_message.py,sha256=m2Lj8plDuJ8QatY19oOSNE77nxr-WwPE4csBPoAIugY,2846
 truefoundry/ml/autogen/client/models/authorize_user_for_model_request_dto.py,sha256=G00T6eroxZrDw782-UBCQxlkwkxyAPfmvd9QNcMiZDI,2017
 truefoundry/ml/autogen/client/models/authorize_user_for_model_version_request_dto.py,sha256=Cy_7YVK47WMgweWbrSzR8-pb3B0WotriQWrKYdcxhvk,2070
-truefoundry/ml/autogen/client/models/blob_storage_reference.py,sha256=UcfYHxRyJyDMge90Ln9Agm9ynbddVjOvGonX6VeKFKo,2995
+truefoundry/ml/autogen/client/models/blob_storage_reference.py,sha256=A6rATKEkYcSSShSInrJtKxwi8GsRuS3JJiq8xi9WFBc,2996
 truefoundry/ml/autogen/client/models/body_get_search_runs_get.py,sha256=aGu5Ah2QB-fnke4joOSAnhD5aO0BFGP3dzZcvSNSe-s,2064
-truefoundry/ml/autogen/client/models/chat_prompt.py,sha256=PFpvfDGWLSUWYTHEFAOtScJeTYjUUziQD_Meii2D0gw,5974
+truefoundry/ml/autogen/client/models/chat_prompt.py,sha256=Yq-DP0ur-ECcxcZH5HiKLcNdWtaTNS_WOdA1nGY3lRE,5975
 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
@@ -199,10 +199,8 @@ 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_artifact_source.py,sha256=s7cDavSpmlCblMVV1DjSWur258YHYPxVjWze3uQTjXY,2360
-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/external_artifact_source.py,sha256=h94vLz48yQg5l86V17Ve9PuwMnmkrLJxeDlX_tFOJ48,2361
+truefoundry/ml/autogen/client/models/fast_ai_framework.py,sha256=GwVoMbIzYhr4srFXyFb565qq-_Q-7SkUsjT7zlRGz2k,2171
 truefoundry/ml/autogen/client/models/file_info_dto.py,sha256=7oc8venicsFVk8zT9wHNhHnZGtFkFlqimFnS7ozGL9k,2156
 truefoundry/ml/autogen/client/models/finalize_artifact_version_request_dto.py,sha256=tkPcNjI0_qhX-D3m6AXqLzdmo9_K0tFBtbPDeg0Zo7U,3818
 truefoundry/ml/autogen/client/models/framework.py,sha256=W_KrqH1qKRvlrssCCWFIlmPaeAw0eiM6VqS3pbdlMXU,14720
@@ -218,16 +216,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/gluon_framework.py,sha256=T9ppm2DsubHcAKsKDDaeFjG88tJix6bdduLCfNpYeac,2150
+truefoundry/ml/autogen/client/models/h2_o_framework.py,sha256=5X-0Xbt4tpecV_q6WwGqxGDvfT6Bue30XW0SiSJojXE,2126
 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_content_part.py,sha256=m_LE5XqD_UD0f8jtWt0adRgz_FoLEvgZbCDTSYORnCc,2717
 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/keras_framework.py,sha256=sBKDT4Dzkvr3Rf7c8YKRbtzaO_KrPNsW8tdWI_n701Q,2150
 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/light_gbm_framework.py,sha256=lblWmFUORAKQr4AUynrktlxYNKDEXh2TRrERo40MROk,2195
 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
@@ -260,21 +258,19 @@ truefoundry/ml/autogen/client/models/mime_type.py,sha256=A-N5-Mof_IyHWguXmG67k9w
 truefoundry/ml/autogen/client/models/model_configuration.py,sha256=XKhwqO8caQ8w9XRCdb_pAl-84cd_PWkEutjLNSby6hQ,3861
 truefoundry/ml/autogen/client/models/model_dto.py,sha256=TUO74MDqe8XCVJBB2O7TiqiRON8qbE7gi8LNA3uWjYA,4685
 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_dto.py,sha256=9Mh4d8V33gEGYr4sys5lYYDA3bHKajR3oAZLZMAy_io,7094
+truefoundry/ml/autogen/client/models/model_version_manifest.py,sha256=N5N1tOEc98RbQbXKpVQV_Na8RfZHJZfz67hqWcGRv_4,4263
 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/onnx_framework.py,sha256=_YoLLcObmeODnHullwJ1D4J8tOp3k5tRr43VkSDUIrw,2138
 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/paddle_framework.py,sha256=yLVdBTV2Kl_QZbCFE6u00GB-BVMsZrTeRedEGoHpEmU,2177
 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/py_torch_framework.py,sha256=tJXPH9m2gOccQgSK4fcWNRu0ay3r9RNvTz1eyKc-1UA,2183
 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
@@ -289,37 +285,42 @@ 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/sklearn_framework.py,sha256=8-GvfwN5fHVDiOXPvVa9rBeG0VEavwwTsfEp3Nl0uO8,2187
+truefoundry/ml/autogen/client/models/sklearn_framework.py,sha256=duOKdvUfGlcY9-HueGj8o3mknuMMe1A2-gwPKo6e8As,2188
 truefoundry/ml/autogen/client/models/source.py,sha256=ZG2-3oIs_M8mZNPjoIRPzhHqlm63vLCVQC9idUxQ0XU,5180
 truefoundry/ml/autogen/client/models/source1.py,sha256=io35b9uVzqSvOHw6RlCXxM30thN-vHm1upaSiN2i0II,5183
-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/spa_cy_framework.py,sha256=Kyrvhdalnl1uQLoPMGMry2PuFGCcQ8AGp4Cx4B9AGXQ,2150
+truefoundry/ml/autogen/client/models/stats_models_framework.py,sha256=p6isf4NlvzL_-WO-YVNclG6i40FP890PSAKQ-VmHgA4,2231
 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/tensor_flow_framework.py,sha256=lvABZqmVED7Hg7aDZ7bovN6usG_TZCWI4B8Ew6NAymw,2218
+truefoundry/ml/autogen/client/models/system_message.py,sha256=2sPSNrDuz5elg_AG009CTe0E3HOFxAXtRKCK_Xp5CPw,2802
+truefoundry/ml/autogen/client/models/tensor_flow_framework.py,sha256=jsAtqm2rMuiQyR7vVNDTLXDqpIPKsXEOH2WLRGE-POw,2219
 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=0lbFboK0hfHxZGLRiLAhX0imn-EAJHunFWDORF6-zSY,3203
+truefoundry/ml/autogen/client/models/text_content_part.py,sha256=IgJ5exYltwJdJUBdOxMr3VtjF6x9qbyA_4mTPDZ4QCY,2493
+truefoundry/ml/autogen/client/models/transformers_framework.py,sha256=NNsTWMXmQyk5pCFowAIIgIs-_Iaht05RK-8-fUcQk18,3204
 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/true_foundry_artifact_source.py,sha256=Qwtowy8XSqMgQQ9yEk9AUL1ghaETLbwQW9sGNQfh6D0,2456
+truefoundry/ml/autogen/client/models/true_foundry_artifact_source.py,sha256=LihHK1hS5sivJ6EtV1TsYU9ATYgSNDhg10Vhq6ypEG4,2457
 truefoundry/ml/autogen/client/models/update_artifact_version_request_dto.py,sha256=nBcEe5aCIvqKz_iIeegkDKsOedSTq6-ahlC_GAwxkO4,2766
 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_model_version_request_dto.py,sha256=GeEprpAgPvCnQe5yjmpofL7WBE0EhEHtFPIwj4jFCI4,2907
 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/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/user_message.py,sha256=3TEM8qH_zT3dgM197bl44uV7m20IWn6sWeLGfStsN90,2784
 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/models/xg_boost_framework.py,sha256=kC5P2rCw4Wp9v2Vok8X6y_kONznJMohrzLIfN6J17lY,2183
 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/client_README.md,sha256=3qPxCLIRf16kGgRP-cXgmpr3KutjLTddIdKUQpJPTpg,35393
+truefoundry/ml/autogen/entities/artifacts.py,sha256=EbGDmD0npfa5BSHau67Y8-dSk6qPJJlJX3I3N4kqAgA,17326
+truefoundry/ml/autogen/models/__init__.py,sha256=--TGRea9SQBWFfwtcl3ekb1XGfFTdEkQGSG8a2SJ60I,187
+truefoundry/ml/autogen/models/exceptions.py,sha256=q3n7FGBrg_hUy1UyoefhMnhcXUAaqXlLIGHoOVzn_d8,1390
+truefoundry/ml/autogen/models/schema.py,sha256=IhpO9qbygLqEamP3NIt3m90SseJXCOm1ZTqNbNbW-M0,55772
+truefoundry/ml/autogen/models/signature.py,sha256=EppDGjws10CTatxcTBTuT-b3jJX9lmR6ZX98eC6PB5w,4942
+truefoundry/ml/autogen/models/utils.py,sha256=Z52SnAF7UL7uluNtA_VXuB9iPju-q9bpTTFJnUh0ZTQ,24289
 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 +340,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 +352,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 +375,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.0rc6.dist-info/METADATA,sha256=uBUgaLgGtMVJfYQnnyciAnXSX-X5Qf1PV1jt5ILZNR8,3102
+truefoundry-0.5.0rc6.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+truefoundry-0.5.0rc6.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.5.0rc6.dist-info/RECORD,,

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

@@ -1,68 +0,0 @@
-# coding: utf-8
-
-"""
-FastAPI
-
-No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
-
-The version of the OpenAPI document: 0.1.0
-Generated by OpenAPI Generator (https://openapi-generator.tech)
-
-Do not edit the class manually.
-"""  # noqa: E501
-
-from __future__ import annotations
-
-import json
-import pprint
-import re  # noqa: F401
-from typing import Optional
-
-from truefoundry.ml.autogen.client.models.feature_value_type import FeatureValueType
-from truefoundry.pydantic_v1 import BaseModel, StrictStr
-
-
-class FeatureDto(BaseModel):
-    """
-    FeatureDto
-    """
-
-    name: Optional[StrictStr] = None
-    type: Optional[FeatureValueType] = None
-    __properties = ["name", "type"]
-
-    class Config:
-        """Pydantic configuration"""
-
-        allow_population_by_field_name = True
-        validate_assignment = True
-
-    def to_str(self) -> str:
-        """Returns the string representation of the model using alias"""
-        return pprint.pformat(self.dict(by_alias=True))
-
-    def to_json(self) -> str:
-        """Returns the JSON representation of the model using alias"""
-        return json.dumps(self.to_dict())
-
-    @classmethod
-    def from_json(cls, json_str: str) -> FeatureDto:
-        """Create an instance of FeatureDto from a JSON string"""
-        return cls.from_dict(json.loads(json_str))
-
-    def to_dict(self):
-        """Returns the dictionary representation of the model using alias"""
-        _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
-        return _dict
-
-    @classmethod
-    def from_dict(cls, obj: dict) -> FeatureDto:
-        """Create an instance of FeatureDto from a dict"""
-        if obj is None:
-            return None
-
-        if not isinstance(obj, dict):
-            return FeatureDto.parse_obj(obj)
-
-        _obj = FeatureDto.parse_obj({"name": obj.get("name"), "type": obj.get("type")})
-        return _obj

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

@@ -1,35 +0,0 @@
-# coding: utf-8
-
-"""
-FastAPI
-
-No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
-
-The version of the OpenAPI document: 0.1.0
-Generated by OpenAPI Generator (https://openapi-generator.tech)
-
-Do not edit the class manually.
-"""  # noqa: E501
-
-import json
-import re  # noqa: F401
-
-from aenum import Enum
-
-
-class FeatureValueType(str, Enum):
-    """
-    An enumeration.
-    """
-
-    """
-    allowed enum values
-    """
-    INT = "int"
-    FLOAT = "float"
-    STRING = "string"
-
-    @classmethod
-    def from_json(cls, json_str: str) -> FeatureValueType:
-        """Create an instance of FeatureValueType from a JSON string"""
-        return FeatureValueType(json.loads(json_str))

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

@@ -1,85 +0,0 @@
-# coding: utf-8
-
-"""
-FastAPI
-
-No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
-
-The version of the OpenAPI document: 0.1.0
-Generated by OpenAPI Generator (https://openapi-generator.tech)
-
-Do not edit the class manually.
-"""  # noqa: E501
-
-from __future__ import annotations
-
-import json
-import pprint
-import re  # noqa: F401
-from typing import Optional
-
-from truefoundry.ml.autogen.client.models.feature_dto import FeatureDto
-from truefoundry.ml.autogen.client.models.prediction_type import PredictionType
-from truefoundry.pydantic_v1 import BaseModel, conlist
-
-
-class ModelSchemaDto(BaseModel):
-    """
-    ModelSchemaDto
-    """
-
-    features: Optional[conlist(FeatureDto)] = None
-    prediction: Optional[PredictionType] = None
-    __properties = ["features", "prediction"]
-
-    class Config:
-        """Pydantic configuration"""
-
-        allow_population_by_field_name = True
-        validate_assignment = True
-
-    def to_str(self) -> str:
-        """Returns the string representation of the model using alias"""
-        return pprint.pformat(self.dict(by_alias=True))
-
-    def to_json(self) -> str:
-        """Returns the JSON representation of the model using alias"""
-        return json.dumps(self.to_dict())
-
-    @classmethod
-    def from_json(cls, json_str: str) -> ModelSchemaDto:
-        """Create an instance of ModelSchemaDto from a JSON string"""
-        return cls.from_dict(json.loads(json_str))
-
-    def to_dict(self):
-        """Returns the dictionary representation of the model using alias"""
-        _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
-        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of each item in features (list)
-        _items = []
-        if self.features:
-            for _item in self.features:
-                if _item:
-                    _items.append(_item.to_dict())
-            _dict["features"] = _items
-        return _dict
-
-    @classmethod
-    def from_dict(cls, obj: dict) -> ModelSchemaDto:
-        """Create an instance of ModelSchemaDto from a dict"""
-        if obj is None:
-            return None
-
-        if not isinstance(obj, dict):
-            return ModelSchemaDto.parse_obj(obj)
-
-        _obj = ModelSchemaDto.parse_obj(
-            {
-                "features": [
-                    FeatureDto.from_dict(_item) for _item in obj.get("features")
-                ]
-                if obj.get("features") is not None
-                else None,
-                "prediction": obj.get("prediction"),
-            }
-        )
-        return _obj

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

@@ -1,34 +0,0 @@
-# coding: utf-8
-
-"""
-FastAPI
-
-No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
-
-The version of the OpenAPI document: 0.1.0
-Generated by OpenAPI Generator (https://openapi-generator.tech)
-
-Do not edit the class manually.
-"""  # noqa: E501
-
-import json
-import re  # noqa: F401
-
-from aenum import Enum
-
-
-class PredictionType(str, Enum):
-    """
-    An enumeration.
-    """
-
-    """
-    allowed enum values
-    """
-    CATEGORICAL = "categorical"
-    NUMERIC = "numeric"
-
-    @classmethod
-    def from_json(cls, json_str: str) -> PredictionType:
-        """Create an instance of PredictionType from a JSON string"""
-        return PredictionType(json.loads(json_str))