truefoundry 0.3.3__py3-none-any.whl → 0.4.0.dev1__py3-none-any.whl

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

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

truefoundry/ml/autogen/client/models/create_artifact_version_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 truefoundry.ml.autogen.client.models.artifact_type import ArtifactType
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr
+
+
+class CreateArtifactVersionRequestDto(BaseModel):
+    """
+    CreateArtifactVersionRequestDto
+    """
+
+    experiment_id: StrictInt = Field(...)
+    name: StrictStr = Field(...)
+    artifact_type: ArtifactType = Field(...)
+    __properties = ["experiment_id", "name", "artifact_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) -> CreateArtifactVersionRequestDto:
+        """Create an instance of CreateArtifactVersionRequestDto 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) -> CreateArtifactVersionRequestDto:
+        """Create an instance of CreateArtifactVersionRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return CreateArtifactVersionRequestDto.parse_obj(obj)
+
+        _obj = CreateArtifactVersionRequestDto.parse_obj(
+            {
+                "experiment_id": obj.get("experiment_id"),
+                "name": obj.get("name"),
+                "artifact_type": obj.get("artifact_type"),
+            }
+        )
+        return _obj

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

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

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

@@ -0,0 +1,76 @@
+# 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.pydantic_v1 import BaseModel, Field, StrictStr
+
+
+class CreateDatasetRequestDto(BaseModel):
+    """
+    CreateDatasetRequestDto
+    """
+
+    name: StrictStr = Field(...)
+    description: Optional[StrictStr] = None
+    experiment_id: Optional[StrictStr] = None
+    dataset_metadata: Optional[Dict[str, Any]] = None
+    __properties = ["name", "description", "experiment_id", "dataset_metadata"]
+
+    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) -> CreateDatasetRequestDto:
+        """Create an instance of CreateDatasetRequestDto 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) -> CreateDatasetRequestDto:
+        """Create an instance of CreateDatasetRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return CreateDatasetRequestDto.parse_obj(obj)
+
+        _obj = CreateDatasetRequestDto.parse_obj(
+            {
+                "name": obj.get("name"),
+                "description": obj.get("description"),
+                "experiment_id": obj.get("experiment_id"),
+                "dataset_metadata": obj.get("dataset_metadata"),
+            }
+        )
+        return _obj

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

@@ -0,0 +1,94 @@
+# 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.experiment_tag_dto import ExperimentTagDto
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, conlist
+
+
+class CreateExperimentRequestDto(BaseModel):
+    """
+    CreateExperimentRequestDto
+    """
+
+    name: StrictStr = Field(...)
+    tags: Optional[conlist(ExperimentTagDto)] = None
+    description: Optional[StrictStr] = None
+    privacy_type: Optional[StrictStr] = None
+    storage_integration_fqn: Optional[StrictStr] = None
+    __properties = [
+        "name",
+        "tags",
+        "description",
+        "privacy_type",
+        "storage_integration_fqn",
+    ]
+
+    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) -> CreateExperimentRequestDto:
+        """Create an instance of CreateExperimentRequestDto 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) -> CreateExperimentRequestDto:
+        """Create an instance of CreateExperimentRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return CreateExperimentRequestDto.parse_obj(obj)
+
+        _obj = CreateExperimentRequestDto.parse_obj(
+            {
+                "name": obj.get("name"),
+                "tags": [ExperimentTagDto.from_dict(_item) for _item in obj.get("tags")]
+                if obj.get("tags") is not None
+                else None,
+                "description": obj.get("description"),
+                "privacy_type": obj.get("privacy_type"),
+                "storage_integration_fqn": obj.get("storage_integration_fqn"),
+            }
+        )
+        return _obj

truefoundry/ml/autogen/client/models/create_experiment_response_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 CreateExperimentResponseDto(BaseModel):
+    """
+    CreateExperimentResponseDto
+    """
+
+    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) -> CreateExperimentResponseDto:
+        """Create an instance of CreateExperimentResponseDto 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) -> CreateExperimentResponseDto:
+        """Create an instance of CreateExperimentResponseDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return CreateExperimentResponseDto.parse_obj(obj)
+
+        _obj = CreateExperimentResponseDto.parse_obj(
+            {"experiment_id": obj.get("experiment_id")}
+        )
+        return _obj

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

@@ -0,0 +1,95 @@
+# 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.internal_metadata import InternalMetadata
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr
+
+
+class CreateModelVersionRequestDto(BaseModel):
+    """
+    CreateModelVersionRequestDto
+    """
+
+    artifact_version_id: StrictStr = Field(...)
+    description: Optional[StrictStr] = None
+    artifact_metadata: Optional[Dict[str, Any]] = None
+    data_path: Optional[StrictStr] = None
+    step: Optional[StrictInt] = None
+    internal_metadata: Optional[InternalMetadata] = None
+    __properties = [
+        "artifact_version_id",
+        "description",
+        "artifact_metadata",
+        "data_path",
+        "step",
+        "internal_metadata",
+    ]
+
+    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) -> CreateModelVersionRequestDto:
+        """Create an instance of CreateModelVersionRequestDto 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 internal_metadata
+        if self.internal_metadata:
+            _dict["internal_metadata"] = self.internal_metadata.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> CreateModelVersionRequestDto:
+        """Create an instance of CreateModelVersionRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return CreateModelVersionRequestDto.parse_obj(obj)
+
+        _obj = CreateModelVersionRequestDto.parse_obj(
+            {
+                "artifact_version_id": obj.get("artifact_version_id"),
+                "description": obj.get("description"),
+                "artifact_metadata": obj.get("artifact_metadata"),
+                "data_path": obj.get("data_path"),
+                "step": obj.get("step"),
+                "internal_metadata": InternalMetadata.from_dict(
+                    obj.get("internal_metadata")
+                )
+                if obj.get("internal_metadata") is not None
+                else None,
+            }
+        )
+        return _obj

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

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

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

@@ -0,0 +1,79 @@
+# 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.multi_part_upload_dto import (
+    MultiPartUploadDto,
+)
+from truefoundry.pydantic_v1 import BaseModel, Field
+
+
+class CreateMultiPartUploadForDatasetResponseDto(BaseModel):
+    """
+    CreateMultiPartUploadForDatasetResponseDto
+    """
+
+    multipart_upload: MultiPartUploadDto = Field(...)
+    __properties = ["multipart_upload"]
+
+    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) -> CreateMultiPartUploadForDatasetResponseDto:
+        """Create an instance of CreateMultiPartUploadForDatasetResponseDto 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 multipart_upload
+        if self.multipart_upload:
+            _dict["multipart_upload"] = self.multipart_upload.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> CreateMultiPartUploadForDatasetResponseDto:
+        """Create an instance of CreateMultiPartUploadForDatasetResponseDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return CreateMultiPartUploadForDatasetResponseDto.parse_obj(obj)
+
+        _obj = CreateMultiPartUploadForDatasetResponseDto.parse_obj(
+            {
+                "multipart_upload": MultiPartUploadDto.from_dict(
+                    obj.get("multipart_upload")
+                )
+                if obj.get("multipart_upload") is not None
+                else None
+            }
+        )
+        return _obj