truefoundry 0.5.1rc1__py3-none-any.whl → 0.5.1rc2__py3-none-any.whl

truefoundry/ml/__init__.py

@@ -1,5 +1,4 @@
 from truefoundry.ml.autogen.client.models import LibraryName, ModelVersionEnvironment
-from truefoundry.ml.autogen.models import infer_signature
 from truefoundry.ml.enums import (
     DataSlice,
     FileFormat,
@@ -17,6 +16,7 @@ from truefoundry.ml.log_types.artifacts.artifact import (
 from truefoundry.ml.log_types.artifacts.dataset import DataDirectory, DataDirectoryPath
 from truefoundry.ml.log_types.artifacts.model import (
     ModelVersion,
+    infer_signature,
 )
 from truefoundry.ml.logger import init_logger
 from truefoundry.ml.mlfoundry_api import get_client

truefoundry/ml/log_types/artifacts/model.py

@@ -54,6 +54,9 @@ from truefoundry.ml.session import _get_api_client
 from truefoundry.pydantic_v1 import BaseModel, Extra
 
 if TYPE_CHECKING:
+    import numpy as np
+    import pandas as pd
+
     from truefoundry.ml.mlfoundry_run import MlFoundryRun
 
 
@@ -624,3 +627,18 @@ def _log_model_version(  # noqa: C901
         )
     )
     return ModelVersion.from_fqn(fqn=artifact_version_response.artifact_version.fqn)
+
+
+def infer_signature(
+    model_input: Any = None,
+    model_output: Optional[
+        Union["pd.DataFrame", "np.ndarray", Dict[str, "np.ndarray"]]
+    ] = None,
+    params: Optional[Dict[str, Any]] = None,
+):
+    # TODO: Importing this globally causes hard dependencies on some libraries like pandas
+    from truefoundry.ml.autogen.models import infer_signature as _infer_signature
+
+    return _infer_signature(
+        model_input=model_input, model_output=model_output, params=params
+    )

{truefoundry-0.5.1rc1.dist-info → truefoundry-0.5.1rc2.dist-info}/METADATA

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

{truefoundry-0.5.1rc1.dist-info → truefoundry-0.5.1rc2.dist-info}/RECORD

@@ -107,7 +107,7 @@ truefoundry/deploy/v2/lib/models.py,sha256=pSolLMTArDuYpeNsmeeS5DWliloN_iCDfZSpR
 truefoundry/deploy/v2/lib/patched_models.py,sha256=NTU8J_CwdvEuF9zNXwFyN3suOp_196Wrm75DDy5qcXE,14184
 truefoundry/deploy/v2/lib/source.py,sha256=VHKuFREiixUP40D3Mrz-TA70spu1M0RbCzl--qwlFaY,9263
 truefoundry/logger.py,sha256=u-YCNjg5HBwE70uQcpjIG64Ghos-K2ulTWaxC03BSj4,714
-truefoundry/ml/__init__.py,sha256=qfYk5FVNDl50_RJBvIjfzQakOQ4HHLH8oVYmb4SOOh0,1963
+truefoundry/ml/__init__.py,sha256=aPrCvFWkb_MGMliT3rBaF4CBcF0AdaymC9DFuK5J7DM,1926
 truefoundry/ml/artifact/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/ml/artifact/truefoundry_artifact_repo.py,sha256=FksxhUpRHb9pgWZmAB16DhXqkAL6UIAPA1S3RJUApQU,46201
 truefoundry/ml/autogen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -331,8 +331,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=QJqqgDbflTfsmdsNuV30HvRb0fnaP2UaTrLLqL7jXGM,23312
-truefoundry/ml/log_types/artifacts/model_extras.py,sha256=TIE73bLKfwIVzNiVcjmaZ841A70BHBwu4XAM6ZAQRFI,1045
+truefoundry/ml/log_types/artifacts/model.py,sha256=BqZPUt3rKYdlceZ35NNU6a31m0FVEKYbrHQb9tIe8wg,23854
 truefoundry/ml/log_types/artifacts/utils.py,sha256=yhIZYe3Y-J0rG0Wcwql_Qjxe9hlefFbXUfRFqg4E9cM,7502
 truefoundry/ml/log_types/image/__init__.py,sha256=fcOq8yQnNj1rkLcPeIjLXBpdA1WIeiPsXOlAAvMxx7M,76
 truefoundry/ml/log_types/image/constants.py,sha256=wLtGEOA4T5fZHSlOXPuNDLX3lpbCtwlvGKPFk_1fah0,255
@@ -366,7 +365,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.1rc1.dist-info/METADATA,sha256=nz3SDzyjyfTrEYBTeMRUAUAvOrtcZewEmKq-AKZwp58,2887
-truefoundry-0.5.1rc1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-truefoundry-0.5.1rc1.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.5.1rc1.dist-info/RECORD,,
+truefoundry-0.5.1rc2.dist-info/METADATA,sha256=vL0TH3dHp17F31KFqk_0vWHF2xefUVPUJHN54CsI7MA,2887
+truefoundry-0.5.1rc2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+truefoundry-0.5.1rc2.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.5.1rc2.dist-info/RECORD,,

truefoundry/ml/log_types/artifacts/model_extras.py

@@ -1,48 +0,0 @@
-import enum
-from typing import List, Type, TypeVar
-
-from truefoundry.ml.autogen.client import (  # type: ignore[attr-defined]
-    ModelSchemaDto,
-)
-from truefoundry.ml.exceptions import MlFoundryException
-from truefoundry.pydantic_v1 import BaseModel
-
-T = TypeVar("T")
-
-
-class BaseEnum(enum.Enum):
-    @classmethod
-    def values(cls: Type[T]) -> List[T]:
-        return [member.value for member in cls]
-
-    @classmethod
-    def _missing_(cls: Type[T], value: object):
-        raise MlFoundryException(
-            f"Unknown value for type {cls.__name__}: {value}", status_code=400
-        )
-
-
-@enum.unique
-class CustomMetricValueType(str, BaseEnum):
-    FLOAT = "float"
-
-
-@enum.unique
-class CustomMetricType(str, BaseEnum):
-    METRIC = "metric"
-    PROJECTION = "projection"
-
-
-class CustomMetric(BaseModel):
-    class Config:
-        validate_assignment = True
-        use_enum_values = True
-        extra = "allow"
-
-    name: str
-    value_type: CustomMetricValueType
-    type: CustomMetricType
-
-
-class ModelSchema(ModelSchemaDto):
-    pass