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

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

@@ -0,0 +1,35 @@
+# 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
+
+import json
+import re  # noqa: F401
+
+from aenum import Enum
+
+
+class FeatureValueType(str, Enum):
+    """
+    An enumeration.
+    """
+
+    """
+    allowed enum values
+    """
+    INT = "int"
+    FLOAT = "float"
+    STRING = "string"
+
+    @classmethod
+    def from_json(cls, json_str: str) -> FeatureValueType:
+        """Create an instance of FeatureValueType from a JSON string"""
+        return FeatureValueType(json.loads(json_str))

truefoundry/ml/autogen/client/models/file_info_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 Optional
+
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictBool, StrictInt, StrictStr
+
+
+class FileInfoDto(BaseModel):
+    """
+    FileInfoDto
+    """
+
+    path: StrictStr = Field(...)
+    is_dir: StrictBool = Field(...)
+    file_size: Optional[StrictInt] = None
+    signed_url: Optional[StrictStr] = None
+    __properties = ["path", "is_dir", "file_size", "signed_url"]
+
+    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) -> FileInfoDto:
+        """Create an instance of FileInfoDto 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) -> FileInfoDto:
+        """Create an instance of FileInfoDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return FileInfoDto.parse_obj(obj)
+
+        _obj = FileInfoDto.parse_obj(
+            {
+                "path": obj.get("path"),
+                "is_dir": obj.get("is_dir"),
+                "file_size": obj.get("file_size"),
+                "signed_url": obj.get("signed_url"),
+            }
+        )
+        return _obj

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

@@ -0,0 +1,101 @@
+# 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 FinalizeArtifactVersionRequestDto(BaseModel):
+    """
+    FinalizeArtifactVersionRequestDto
+    """
+
+    id: StrictStr = Field(...)
+    run_uuid: Optional[StrictStr] = None
+    description: Optional[StrictStr] = None
+    artifact_metadata: Optional[Dict[str, Any]] = None
+    internal_metadata: Optional[InternalMetadata] = None
+    data_path: Optional[StrictStr] = None
+    step: Optional[StrictInt] = None
+    artifact_size: Optional[StrictInt] = None
+    __properties = [
+        "id",
+        "run_uuid",
+        "description",
+        "artifact_metadata",
+        "internal_metadata",
+        "data_path",
+        "step",
+        "artifact_size",
+    ]
+
+    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) -> FinalizeArtifactVersionRequestDto:
+        """Create an instance of FinalizeArtifactVersionRequestDto 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) -> FinalizeArtifactVersionRequestDto:
+        """Create an instance of FinalizeArtifactVersionRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return FinalizeArtifactVersionRequestDto.parse_obj(obj)
+
+        _obj = FinalizeArtifactVersionRequestDto.parse_obj(
+            {
+                "id": obj.get("id"),
+                "run_uuid": obj.get("run_uuid"),
+                "description": obj.get("description"),
+                "artifact_metadata": obj.get("artifact_metadata"),
+                "internal_metadata": InternalMetadata.from_dict(
+                    obj.get("internal_metadata")
+                )
+                if obj.get("internal_metadata") is not None
+                else None,
+                "data_path": obj.get("data_path"),
+                "step": obj.get("step"),
+                "artifact_size": obj.get("artifact_size"),
+            }
+        )
+        return _obj

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

@@ -0,0 +1,88 @@
+# 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_dto import ExperimentDto
+from truefoundry.ml.autogen.client.models.run_info_dto import RunInfoDto
+from truefoundry.pydantic_v1 import BaseModel, Field, conlist
+
+
+class GetExperimentResponseDto(BaseModel):
+    """
+    GetExperimentResponseDto
+    """
+
+    experiment: ExperimentDto = Field(...)
+    runs: Optional[conlist(RunInfoDto)] = None
+    __properties = ["experiment", "runs"]
+
+    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) -> GetExperimentResponseDto:
+        """Create an instance of GetExperimentResponseDto 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()
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of each item in runs (list)
+        _items = []
+        if self.runs:
+            for _item in self.runs:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict["runs"] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> GetExperimentResponseDto:
+        """Create an instance of GetExperimentResponseDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return GetExperimentResponseDto.parse_obj(obj)
+
+        _obj = GetExperimentResponseDto.parse_obj(
+            {
+                "experiment": ExperimentDto.from_dict(obj.get("experiment"))
+                if obj.get("experiment") is not None
+                else None,
+                "runs": [RunInfoDto.from_dict(_item) for _item in obj.get("runs")]
+                if obj.get("runs") is not None
+                else None,
+            }
+        )
+        return _obj

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

truefoundry/ml/autogen/client/models/get_metric_history_response.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.metric_dto import MetricDto
+from truefoundry.pydantic_v1 import BaseModel, Field, conlist
+
+
+class GetMetricHistoryResponse(BaseModel):
+    """
+    GetMetricHistoryResponse
+    """
+
+    metrics: conlist(MetricDto) = Field(...)
+    __properties = ["metrics"]
+
+    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) -> GetMetricHistoryResponse:
+        """Create an instance of GetMetricHistoryResponse 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 metrics (list)
+        _items = []
+        if self.metrics:
+            for _item in self.metrics:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict["metrics"] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> GetMetricHistoryResponse:
+        """Create an instance of GetMetricHistoryResponse from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return GetMetricHistoryResponse.parse_obj(obj)
+
+        _obj = GetMetricHistoryResponse.parse_obj(
+            {
+                "metrics": [MetricDto.from_dict(_item) for _item in obj.get("metrics")]
+                if obj.get("metrics") is not None
+                else None
+            }
+        )
+        return _obj

truefoundry/ml/autogen/client/models/get_signed_url_for_dataset_write_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, conlist
+
+
+class GetSignedURLForDatasetWriteRequestDto(BaseModel):
+    """
+    GetSignedURLForDatasetWriteRequestDto
+    """
+
+    dataset_fqn: StrictStr = Field(...)
+    paths: conlist(StrictStr) = Field(...)
+    __properties = ["dataset_fqn", "paths"]
+
+    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) -> GetSignedURLForDatasetWriteRequestDto:
+        """Create an instance of GetSignedURLForDatasetWriteRequestDto 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) -> GetSignedURLForDatasetWriteRequestDto:
+        """Create an instance of GetSignedURLForDatasetWriteRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return GetSignedURLForDatasetWriteRequestDto.parse_obj(obj)
+
+        _obj = GetSignedURLForDatasetWriteRequestDto.parse_obj(
+            {"dataset_fqn": obj.get("dataset_fqn"), "paths": obj.get("paths")}
+        )
+        return _obj

truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_read_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, conlist
+
+
+class GetSignedURLsForArtifactVersionReadRequestDto(BaseModel):
+    """
+    GetSignedURLsForArtifactVersionReadRequestDto
+    """
+
+    id: StrictStr = Field(...)
+    paths: conlist(StrictStr) = Field(...)
+    __properties = ["id", "paths"]
+
+    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) -> GetSignedURLsForArtifactVersionReadRequestDto:
+        """Create an instance of GetSignedURLsForArtifactVersionReadRequestDto 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) -> GetSignedURLsForArtifactVersionReadRequestDto:
+        """Create an instance of GetSignedURLsForArtifactVersionReadRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return GetSignedURLsForArtifactVersionReadRequestDto.parse_obj(obj)
+
+        _obj = GetSignedURLsForArtifactVersionReadRequestDto.parse_obj(
+            {"id": obj.get("id"), "paths": obj.get("paths")}
+        )
+        return _obj

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

@@ -0,0 +1,81 @@
+# 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.signed_url_dto import SignedURLDto
+from truefoundry.pydantic_v1 import BaseModel, Field, conlist
+
+
+class GetSignedURLsForArtifactVersionReadResponseDto(BaseModel):
+    """
+    GetSignedURLsForArtifactVersionReadResponseDto
+    """
+
+    signed_urls: conlist(SignedURLDto) = Field(...)
+    __properties = ["signed_urls"]
+
+    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) -> GetSignedURLsForArtifactVersionReadResponseDto:
+        """Create an instance of GetSignedURLsForArtifactVersionReadResponseDto 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 signed_urls (list)
+        _items = []
+        if self.signed_urls:
+            for _item in self.signed_urls:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict["signed_urls"] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> GetSignedURLsForArtifactVersionReadResponseDto:
+        """Create an instance of GetSignedURLsForArtifactVersionReadResponseDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return GetSignedURLsForArtifactVersionReadResponseDto.parse_obj(obj)
+
+        _obj = GetSignedURLsForArtifactVersionReadResponseDto.parse_obj(
+            {
+                "signed_urls": [
+                    SignedURLDto.from_dict(_item) for _item in obj.get("signed_urls")
+                ]
+                if obj.get("signed_urls") is not None
+                else None
+            }
+        )
+        return _obj