truefoundry 0.3.2__py3-none-any.whl → 0.4.0.dev0__py3-none-any.whl

truefoundry/ml/autogen/client/models/delete_dataset_request_dto.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 typing import Optional
+
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictBool, StrictStr
+
+
+class DeleteDatasetRequestDto(BaseModel):
+    """
+    DeleteDatasetRequestDto
+    """
+
+    id: StrictStr = Field(...)
+    delete_contents: Optional[StrictBool] = False
+    __properties = ["id", "delete_contents"]
+
+    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) -> DeleteDatasetRequestDto:
+        """Create an instance of DeleteDatasetRequestDto 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) -> DeleteDatasetRequestDto:
+        """Create an instance of DeleteDatasetRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return DeleteDatasetRequestDto.parse_obj(obj)
+
+        _obj = DeleteDatasetRequestDto.parse_obj(
+            {
+                "id": obj.get("id"),
+                "delete_contents": obj.get("delete_contents")
+                if obj.get("delete_contents") is not None
+                else False,
+            }
+        )
+        return _obj

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

@@ -0,0 +1,65 @@
+# 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
+
+
+class DeleteModelVersionRequestDto(BaseModel):
+    """
+    DeleteModelVersionRequestDto
+    """
+
+    id: StrictStr = Field(...)
+    __properties = ["id"]
+
+    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) -> DeleteModelVersionRequestDto:
+        """Create an instance of DeleteModelVersionRequestDto 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) -> DeleteModelVersionRequestDto:
+        """Create an instance of DeleteModelVersionRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return DeleteModelVersionRequestDto.parse_obj(obj)
+
+        _obj = DeleteModelVersionRequestDto.parse_obj({"id": obj.get("id")})
+        return _obj

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

@@ -0,0 +1,65 @@
+# 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
+
+
+class DeleteRunRequest(BaseModel):
+    """
+    DeleteRunRequest
+    """
+
+    run_id: StrictStr = Field(...)
+    __properties = ["run_id"]
+
+    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) -> DeleteRunRequest:
+        """Create an instance of DeleteRunRequest 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) -> DeleteRunRequest:
+        """Create an instance of DeleteRunRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return DeleteRunRequest.parse_obj(obj)
+
+        _obj = DeleteRunRequest.parse_obj({"run_id": obj.get("run_id")})
+        return _obj

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

@@ -0,0 +1,68 @@
+# 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
+
+
+class DeleteTagRequestDto(BaseModel):
+    """
+    DeleteTagRequestDto
+    """
+
+    run_id: StrictStr = Field(...)
+    key: StrictStr = Field(...)
+    __properties = ["run_id", "key"]
+
+    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) -> DeleteTagRequestDto:
+        """Create an instance of DeleteTagRequestDto 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) -> DeleteTagRequestDto:
+        """Create an instance of DeleteTagRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return DeleteTagRequestDto.parse_obj(obj)
+
+        _obj = DeleteTagRequestDto.parse_obj(
+            {"run_id": obj.get("run_id"), "key": obj.get("key")}
+        )
+        return _obj

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

@@ -0,0 +1,127 @@
+# 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 Any, Dict, Optional
+
+from truefoundry.ml.autogen.client.models.experiment_tag_dto import ExperimentTagDto
+from truefoundry.pydantic_v1 import BaseModel, StrictInt, StrictStr, conlist
+
+
+class ExperimentDto(BaseModel):
+    """
+    ExperimentDto
+    """
+
+    experiment_id: Optional[StrictStr] = None
+    name: Optional[StrictStr] = None
+    artifact_location: Optional[StrictStr] = None
+    lifecycle_stage: Optional[StrictStr] = None
+    tags: Optional[conlist(ExperimentTagDto)] = None
+    creator_user_id: Optional[StrictStr] = None
+    num_runs: Optional[StrictInt] = None
+    num_user_collaborators: Optional[StrictInt] = None
+    description: Optional[StrictStr] = None
+    privacy_type: Optional[StrictStr] = None
+    created_at: Optional[StrictInt] = None
+    tenant_name: Optional[StrictStr] = None
+    total_models_count: Optional[StrictInt] = None
+    total_artifacts_count: Optional[StrictInt] = None
+    storage_integration_id: Optional[StrictStr] = None
+    manifest: Optional[Dict[str, Any]] = None
+    __properties = [
+        "experiment_id",
+        "name",
+        "artifact_location",
+        "lifecycle_stage",
+        "tags",
+        "creator_user_id",
+        "num_runs",
+        "num_user_collaborators",
+        "description",
+        "privacy_type",
+        "created_at",
+        "tenant_name",
+        "total_models_count",
+        "total_artifacts_count",
+        "storage_integration_id",
+        "manifest",
+    ]
+
+    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) -> ExperimentDto:
+        """Create an instance of ExperimentDto 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 tags (list)
+        _items = []
+        if self.tags:
+            for _item in self.tags:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict["tags"] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> ExperimentDto:
+        """Create an instance of ExperimentDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return ExperimentDto.parse_obj(obj)
+
+        _obj = ExperimentDto.parse_obj(
+            {
+                "experiment_id": obj.get("experiment_id"),
+                "name": obj.get("name"),
+                "artifact_location": obj.get("artifact_location"),
+                "lifecycle_stage": obj.get("lifecycle_stage"),
+                "tags": [ExperimentTagDto.from_dict(_item) for _item in obj.get("tags")]
+                if obj.get("tags") is not None
+                else None,
+                "creator_user_id": obj.get("creator_user_id"),
+                "num_runs": obj.get("num_runs"),
+                "num_user_collaborators": obj.get("num_user_collaborators"),
+                "description": obj.get("description"),
+                "privacy_type": obj.get("privacy_type"),
+                "created_at": obj.get("created_at"),
+                "tenant_name": obj.get("tenant_name"),
+                "total_models_count": obj.get("total_models_count"),
+                "total_artifacts_count": obj.get("total_artifacts_count"),
+                "storage_integration_id": obj.get("storage_integration_id"),
+                "manifest": obj.get("manifest"),
+            }
+        )
+        return _obj

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

@@ -0,0 +1,67 @@
+# 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
+
+
+class ExperimentIdRequestDto(BaseModel):
+    """
+    ExperimentIdRequestDto
+    """
+
+    experiment_id: StrictStr = Field(...)
+    __properties = ["experiment_id"]
+
+    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) -> ExperimentIdRequestDto:
+        """Create an instance of ExperimentIdRequestDto 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) -> ExperimentIdRequestDto:
+        """Create an instance of ExperimentIdRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return ExperimentIdRequestDto.parse_obj(obj)
+
+        _obj = ExperimentIdRequestDto.parse_obj(
+            {"experiment_id": obj.get("experiment_id")}
+        )
+        return _obj

truefoundry/ml/autogen/client/models/experiment_response_dto.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.ml.autogen.client.models.experiment_dto import ExperimentDto
+from truefoundry.pydantic_v1 import BaseModel, Field
+
+
+class ExperimentResponseDto(BaseModel):
+    """
+    ExperimentResponseDto
+    """
+
+    experiment: ExperimentDto = Field(...)
+    __properties = ["experiment"]
+
+    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) -> ExperimentResponseDto:
+        """Create an instance of ExperimentResponseDto 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 experiment
+        if self.experiment:
+            _dict["experiment"] = self.experiment.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> ExperimentResponseDto:
+        """Create an instance of ExperimentResponseDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return ExperimentResponseDto.parse_obj(obj)
+
+        _obj = ExperimentResponseDto.parse_obj(
+            {
+                "experiment": ExperimentDto.from_dict(obj.get("experiment"))
+                if obj.get("experiment") is not None
+                else None
+            }
+        )
+        return _obj

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

@@ -0,0 +1,69 @@
+# 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.pydantic_v1 import BaseModel, StrictStr
+
+
+class ExperimentTagDto(BaseModel):
+    """
+    ExperimentTagDto
+    """
+
+    key: Optional[StrictStr] = None
+    value: Optional[StrictStr] = None
+    __properties = ["key", "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) -> ExperimentTagDto:
+        """Create an instance of ExperimentTagDto 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) -> ExperimentTagDto:
+        """Create an instance of ExperimentTagDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return ExperimentTagDto.parse_obj(obj)
+
+        _obj = ExperimentTagDto.parse_obj(
+            {"key": obj.get("key"), "value": obj.get("value")}
+        )
+        return _obj

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

@@ -0,0 +1,68 @@
+# 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