truefoundry 0.4.5__py3-none-any.whl → 0.5.0rc1__py3-none-any.whl

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

@@ -0,0 +1,74 @@
+# 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 truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
+
+
+class GluonFramework(BaseModel):
+    """
+    +docs=Gluon framework for the model version +label=Gluon  # noqa: E501
+    """
+
+    type: StrictStr = Field(
+        default=..., description="+label=Type +usage=Type of the framework +value=gluon"
+    )
+    __properties = ["type"]
+
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in ("gluon"):
+            raise ValueError("must be one of enum values ('gluon')")
+        return value
+
+    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) -> GluonFramework:
+        """Create an instance of GluonFramework 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) -> GluonFramework:
+        """Create an instance of GluonFramework from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return GluonFramework.parse_obj(obj)
+
+        _obj = GluonFramework.parse_obj({"type": obj.get("type")})
+        return _obj

truefoundry/ml/autogen/client/models/{upload_model_source.py → h2_o_framework.py}

@@ -20,21 +20,21 @@ import re  # noqa: F401
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 
-class UploadModelSource(BaseModel):
+class H2OFramework(BaseModel):
     """
-    UploadModelSource
+    +docs=H2O framework for the model version +label=H2O  # noqa: E501
     """
 
     type: StrictStr = Field(
-        default=..., description="+label=Type +usage=Type of the source"
+        default=..., description="+label=Type +usage=Type of the framework +value=h2o"
     )
     __properties = ["type"]
 
     @validator("type")
     def type_validate_enum(cls, value):
         """Validates the enum"""
-        if value not in ("upload"):
-            raise ValueError("must be one of enum values ('upload')")
+        if value not in ("h2o"):
+            raise ValueError("must be one of enum values ('h2o')")
         return value
 
     class Config:
@@ -52,8 +52,8 @@ class UploadModelSource(BaseModel):
         return json.dumps(self.to_dict())
 
     @classmethod
-    def from_json(cls, json_str: str) -> UploadModelSource:
-        """Create an instance of UploadModelSource from a JSON string"""
+    def from_json(cls, json_str: str) -> H2OFramework:
+        """Create an instance of H2OFramework from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self):
@@ -62,13 +62,13 @@ class UploadModelSource(BaseModel):
         return _dict
 
     @classmethod
-    def from_dict(cls, obj: dict) -> UploadModelSource:
-        """Create an instance of UploadModelSource from a dict"""
+    def from_dict(cls, obj: dict) -> H2OFramework:
+        """Create an instance of H2OFramework from a dict"""
         if obj is None:
             return None
 
         if not isinstance(obj, dict):
-            return UploadModelSource.parse_obj(obj)
+            return H2OFramework.parse_obj(obj)
 
-        _obj = UploadModelSource.parse_obj({"type": obj.get("type")})
+        _obj = H2OFramework.parse_obj({"type": obj.get("type")})
         return _obj

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

@@ -0,0 +1,74 @@
+# 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 truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
+
+
+class KerasFramework(BaseModel):
+    """
+    +docs=Keras framework for the model version +label=Keras  # noqa: E501
+    """
+
+    type: StrictStr = Field(
+        default=..., description="+label=Type +usage=Type of the framework +value=keras"
+    )
+    __properties = ["type"]
+
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in ("keras"):
+            raise ValueError("must be one of enum values ('keras')")
+        return value
+
+    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) -> KerasFramework:
+        """Create an instance of KerasFramework 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) -> KerasFramework:
+        """Create an instance of KerasFramework from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return KerasFramework.parse_obj(obj)
+
+        _obj = KerasFramework.parse_obj({"type": obj.get("type")})
+        return _obj

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

@@ -0,0 +1,75 @@
+# 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 truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
+
+
+class LightGBMFramework(BaseModel):
+    """
+    +docs=LightGBM framework for the model version +label=LightGBM  # noqa: E501
+    """
+
+    type: StrictStr = Field(
+        default=...,
+        description="+label=Type +usage=Type of the framework +value=lightgbm",
+    )
+    __properties = ["type"]
+
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in ("lightgbm"):
+            raise ValueError("must be one of enum values ('lightgbm')")
+        return value
+
+    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) -> LightGBMFramework:
+        """Create an instance of LightGBMFramework 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) -> LightGBMFramework:
+        """Create an instance of LightGBMFramework from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return LightGBMFramework.parse_obj(obj)
+
+        _obj = LightGBMFramework.parse_obj({"type": obj.get("type")})
+        return _obj

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

@@ -41,7 +41,7 @@ class ModelVersionManifest(BaseModel):
     )
     metadata: Dict[str, Any] = Field(
         default=...,
-        description="+label=Metadata +docs=Metadata for the artifact version +usage=Metadata for the artifact version",
+        description="+label=Metadata +docs=Metadata for the model version +usage=Metadata for the model version +uiType=JsonInput",
     )
     type: Optional[StrictStr] = "model-version"
     source: Source = Field(...)

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

@@ -0,0 +1,74 @@
+# 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 truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
+
+
+class ONNXFramework(BaseModel):
+    """
+    +docs=ONNX framework for the model version +label=ONNX  # noqa: E501
+    """
+
+    type: StrictStr = Field(
+        default=..., description="+label=Type +usage=Type of the framework +value=onnx"
+    )
+    __properties = ["type"]
+
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in ("onnx"):
+            raise ValueError("must be one of enum values ('onnx')")
+        return value
+
+    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) -> ONNXFramework:
+        """Create an instance of ONNXFramework 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) -> ONNXFramework:
+        """Create an instance of ONNXFramework from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return ONNXFramework.parse_obj(obj)
+
+        _obj = ONNXFramework.parse_obj({"type": obj.get("type")})
+        return _obj

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

@@ -0,0 +1,75 @@
+# 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 truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
+
+
+class PaddleFramework(BaseModel):
+    """
+    +docs=PaddlePaddle framework for the model version +label=Paddle  # noqa: E501
+    """
+
+    type: StrictStr = Field(
+        default=...,
+        description="+label=Type +usage=Type of the framework +value=paddle",
+    )
+    __properties = ["type"]
+
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in ("paddle"):
+            raise ValueError("must be one of enum values ('paddle')")
+        return value
+
+    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) -> PaddleFramework:
+        """Create an instance of PaddleFramework 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) -> PaddleFramework:
+        """Create an instance of PaddleFramework from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return PaddleFramework.parse_obj(obj)
+
+        _obj = PaddleFramework.parse_obj({"type": obj.get("type")})
+        return _obj

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

@@ -0,0 +1,75 @@
+# 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 truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
+
+
+class PyTorchFramework(BaseModel):
+    """
+    +docs=PyTorch framework for the model version +label=PyTorch  # noqa: E501
+    """
+
+    type: StrictStr = Field(
+        default=...,
+        description="+label=Type +usage=Type of the framework +value=pytorch",
+    )
+    __properties = ["type"]
+
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in ("pytorch"):
+            raise ValueError("must be one of enum values ('pytorch')")
+        return value
+
+    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) -> PyTorchFramework:
+        """Create an instance of PyTorchFramework 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) -> PyTorchFramework:
+        """Create an instance of PyTorchFramework from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return PyTorchFramework.parse_obj(obj)
+
+        _obj = PyTorchFramework.parse_obj({"type": obj.get("type")})
+        return _obj

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

@@ -0,0 +1,75 @@
+# 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 truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
+
+
+class SklearnFramework(BaseModel):
+    """
+    +docs=Scikit-learn framework for the model version +label=Sklearn  # noqa: E501
+    """
+
+    type: StrictStr = Field(
+        default=...,
+        description="+label=Type +usage=Type of the framework +value=sklearn",
+    )
+    __properties = ["type"]
+
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in ("sklearn"):
+            raise ValueError("must be one of enum values ('sklearn')")
+        return value
+
+    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) -> SklearnFramework:
+        """Create an instance of SklearnFramework 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) -> SklearnFramework:
+        """Create an instance of SklearnFramework from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return SklearnFramework.parse_obj(obj)
+
+        _obj = SklearnFramework.parse_obj({"type": obj.get("type")})
+        return _obj

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

@@ -24,7 +24,6 @@ from truefoundry.ml.autogen.client.models.external_model_source import (
 from truefoundry.ml.autogen.client.models.truefoundry_model_source import (
     TruefoundryModelSource,
 )
-from truefoundry.ml.autogen.client.models.upload_model_source import UploadModelSource
 from truefoundry.pydantic_v1 import (
     BaseModel,
     Field,
@@ -32,28 +31,20 @@ from truefoundry.pydantic_v1 import (
     validator,
 )
 
-SOURCE_ANY_OF_SCHEMAS = [
-    "ExternalModelSource",
-    "TruefoundryModelSource",
-    "UploadModelSource",
-]
+SOURCE_ANY_OF_SCHEMAS = ["ExternalModelSource", "TruefoundryModelSource"]
 
 
 class Source(BaseModel):
     """
-    +label=Artifact Source +docs=Source for the model version +usage=Source for the model version
+    +label=Model Source +usage=Source for the model version +uiType=Group
     """
 
-    # data type: UploadModelSource
-    anyof_schema_1_validator: Optional[UploadModelSource] = None
     # data type: TruefoundryModelSource
-    anyof_schema_2_validator: Optional[TruefoundryModelSource] = None
+    anyof_schema_1_validator: Optional[TruefoundryModelSource] = None
     # data type: ExternalModelSource
-    anyof_schema_3_validator: Optional[ExternalModelSource] = None
+    anyof_schema_2_validator: Optional[ExternalModelSource] = None
     if TYPE_CHECKING:
-        actual_instance: Union[
-            ExternalModelSource, TruefoundryModelSource, UploadModelSource
-        ]
+        actual_instance: Union[ExternalModelSource, TruefoundryModelSource]
     else:
         actual_instance: Any
     any_of_schemas: List[str] = Field(SOURCE_ANY_OF_SCHEMAS, const=True)
@@ -79,14 +70,6 @@ class Source(BaseModel):
     def actual_instance_must_validate_anyof(cls, v):
         instance = Source.construct()
         error_messages = []
-        # validate data type: UploadModelSource
-        if not isinstance(v, UploadModelSource):
-            error_messages.append(
-                f"Error! Input type `{type(v)}` is not `UploadModelSource`"
-            )
-        else:
-            return v
-
         # validate data type: TruefoundryModelSource
         if not isinstance(v, TruefoundryModelSource):
             error_messages.append(
@@ -106,7 +89,7 @@ class Source(BaseModel):
         if error_messages:
             # no match
             raise ValueError(
-                "No match found when setting the actual_instance in Source with anyOf schemas: ExternalModelSource, TruefoundryModelSource, UploadModelSource. Details: "
+                "No match found when setting the actual_instance in Source with anyOf schemas: ExternalModelSource, TruefoundryModelSource. Details: "
                 + ", ".join(error_messages)
             )
         else:
@@ -121,19 +104,13 @@ class Source(BaseModel):
         """Returns the object represented by the json string"""
         instance = Source.construct()
         error_messages = []
-        # anyof_schema_1_validator: Optional[UploadModelSource] = None
-        try:
-            instance.actual_instance = UploadModelSource.from_json(json_str)
-            return instance
-        except (ValidationError, ValueError) as e:
-            error_messages.append(str(e))
-        # anyof_schema_2_validator: Optional[TruefoundryModelSource] = None
+        # anyof_schema_1_validator: Optional[TruefoundryModelSource] = None
         try:
             instance.actual_instance = TruefoundryModelSource.from_json(json_str)
             return instance
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # anyof_schema_3_validator: Optional[ExternalModelSource] = None
+        # anyof_schema_2_validator: Optional[ExternalModelSource] = None
         try:
             instance.actual_instance = ExternalModelSource.from_json(json_str)
             return instance
@@ -143,7 +120,7 @@ class Source(BaseModel):
         if error_messages:
             # no match
             raise ValueError(
-                "No match found when deserializing the JSON string into Source with anyOf schemas: ExternalModelSource, TruefoundryModelSource, UploadModelSource. Details: "
+                "No match found when deserializing the JSON string into Source with anyOf schemas: ExternalModelSource, TruefoundryModelSource. Details: "
                 + ", ".join(error_messages)
             )
         else: