truefoundry 0.5.1rc1__tar.gz → 0.5.1rc2__tar.gz

{truefoundry-0.5.1rc1 → truefoundry-0.5.1rc2}/PKG-INFO

@@ -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 → truefoundry-0.5.1rc2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "truefoundry"
-version = "0.5.1rc1"
+version = "0.5.1rc2"
 description = "Truefoundry CLI"
 authors = ["Abhishek Choudhary <abhishek@truefoundry.com>"]
 readme = "README.md"
@@ -60,6 +60,11 @@ langchain = ">=0.2.0,<0.3.0"
 matplotlib = ">=3.0.0,<4.0.0"
 mypy = "*"
 plotly = ">=5.8.2,<6.0.0"
+pandas = [
+    # This is split into two because poetry will select a version which has no wheel on python >= 3.10 :/
+    { version = ">=1.0.0,<3.0.0", python = "<3.10" },
+    { version = ">=1.4.0,<3.0.0", python = ">=3.10" },
+]
 pre-commit = ">=3.5.0,<4.0.0"
 pytest = ">=8.2.2,<9.0.0"
 pytest-env = ">=1.1.5,<2.0.0"

{truefoundry-0.5.1rc1 → truefoundry-0.5.1rc2}/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-0.5.1rc1 → truefoundry-0.5.1rc2}/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/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