truefoundry 0.3.4rc1__py3-none-any.whl → 0.4.0__py3-none-any.whl

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

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

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

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

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

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

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

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

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

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

@@ -0,0 +1,93 @@
+# 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.artifact_version_status import (
+    ArtifactVersionStatus,
+)
+from truefoundry.pydantic_v1 import BaseModel, StrictBool, StrictInt, StrictStr, conlist
+
+
+class ListModelVersionsRequestDto(BaseModel):
+    """
+    ListModelVersionsRequestDto
+    """
+
+    model_id: Optional[StrictStr] = None
+    statuses: Optional[conlist(ArtifactVersionStatus)] = None
+    max_results: Optional[StrictInt] = None
+    offset: Optional[StrictInt] = None
+    page_token: Optional[StrictStr] = None
+    monitoring_enabled: Optional[StrictBool] = None
+    run_ids: Optional[conlist(StrictStr)] = None
+    __properties = [
+        "model_id",
+        "statuses",
+        "max_results",
+        "offset",
+        "page_token",
+        "monitoring_enabled",
+        "run_ids",
+    ]
+
+    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) -> ListModelVersionsRequestDto:
+        """Create an instance of ListModelVersionsRequestDto 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) -> ListModelVersionsRequestDto:
+        """Create an instance of ListModelVersionsRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return ListModelVersionsRequestDto.parse_obj(obj)
+
+        _obj = ListModelVersionsRequestDto.parse_obj(
+            {
+                "model_id": obj.get("model_id"),
+                "statuses": obj.get("statuses"),
+                "max_results": obj.get("max_results"),
+                "offset": obj.get("offset"),
+                "page_token": obj.get("page_token"),
+                "monitoring_enabled": obj.get("monitoring_enabled"),
+                "run_ids": obj.get("run_ids"),
+            }
+        )
+        return _obj

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

@@ -0,0 +1,89 @@
+# 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, StrictBool, StrictInt, StrictStr
+
+
+class ListModelsRequestDto(BaseModel):
+    """
+    ListModelsRequestDto
+    """
+
+    experiment_id: Optional[StrictStr] = None
+    name: Optional[StrictStr] = None
+    max_results: Optional[StrictInt] = None
+    offset: Optional[StrictInt] = None
+    page_token: Optional[StrictStr] = None
+    monitoring_enabled_only: Optional[StrictBool] = False
+    __properties = [
+        "experiment_id",
+        "name",
+        "max_results",
+        "offset",
+        "page_token",
+        "monitoring_enabled_only",
+    ]
+
+    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) -> ListModelsRequestDto:
+        """Create an instance of ListModelsRequestDto 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) -> ListModelsRequestDto:
+        """Create an instance of ListModelsRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return ListModelsRequestDto.parse_obj(obj)
+
+        _obj = ListModelsRequestDto.parse_obj(
+            {
+                "experiment_id": obj.get("experiment_id"),
+                "name": obj.get("name"),
+                "max_results": obj.get("max_results"),
+                "offset": obj.get("offset"),
+                "page_token": obj.get("page_token"),
+                "monitoring_enabled_only": obj.get("monitoring_enabled_only")
+                if obj.get("monitoring_enabled_only") is not None
+                else False,
+            }
+        )
+        return _obj

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

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