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

truefoundry/ml/autogen/client/models/list_run_artifacts_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.file_info_dto import FileInfoDto
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, conlist
+
+
+class ListRunArtifactsResponseDto(BaseModel):
+    """
+    ListRunArtifactsResponseDto
+    """
+
+    root_uri: Optional[StrictStr] = None
+    files: conlist(FileInfoDto) = Field(...)
+    next_page_token: Optional[StrictStr] = None
+    __properties = ["root_uri", "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) -> ListRunArtifactsResponseDto:
+        """Create an instance of ListRunArtifactsResponseDto 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) -> ListRunArtifactsResponseDto:
+        """Create an instance of ListRunArtifactsResponseDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return ListRunArtifactsResponseDto.parse_obj(obj)
+
+        _obj = ListRunArtifactsResponseDto.parse_obj(
+            {
+                "root_uri": obj.get("root_uri"),
+                "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_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 typing import Optional
+
+from truefoundry.ml.autogen.client.models.run_log_dto import RunLogDto
+from truefoundry.pydantic_v1 import BaseModel, conlist
+
+
+class ListRunLogsResponseDto(BaseModel):
+    """
+    ListRunLogsResponseDto
+    """
+
+    run_logs: Optional[conlist(RunLogDto)] = None
+    __properties = ["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) -> ListRunLogsResponseDto:
+        """Create an instance of ListRunLogsResponseDto 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 run_logs (list)
+        _items = []
+        if self.run_logs:
+            for _item in self.run_logs:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict["run_logs"] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> ListRunLogsResponseDto:
+        """Create an instance of ListRunLogsResponseDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return ListRunLogsResponseDto.parse_obj(obj)
+
+        _obj = ListRunLogsResponseDto.parse_obj(
+            {
+                "run_logs": [
+                    RunLogDto.from_dict(_item) for _item in obj.get("run_logs")
+                ]
+                if obj.get("run_logs") is not None
+                else None
+            }
+        )
+        return _obj

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

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

@@ -0,0 +1,106 @@
+# 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.metric_dto import MetricDto
+from truefoundry.ml.autogen.client.models.param_dto import ParamDto
+from truefoundry.ml.autogen.client.models.run_tag_dto import RunTagDto
+from truefoundry.pydantic_v1 import BaseModel, StrictStr, conlist
+
+
+class LogBatchRequestDto(BaseModel):
+    """
+    LogBatchRequestDto
+    """
+
+    run_id: Optional[StrictStr] = None
+    metrics: Optional[conlist(MetricDto)] = None
+    params: Optional[conlist(ParamDto)] = None
+    tags: Optional[conlist(RunTagDto)] = None
+    __properties = ["run_id", "metrics", "params", "tags"]
+
+    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) -> LogBatchRequestDto:
+        """Create an instance of LogBatchRequestDto 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
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of each item in params (list)
+        _items = []
+        if self.params:
+            for _item in self.params:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict["params"] = _items
+        # 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) -> LogBatchRequestDto:
+        """Create an instance of LogBatchRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return LogBatchRequestDto.parse_obj(obj)
+
+        _obj = LogBatchRequestDto.parse_obj(
+            {
+                "run_id": obj.get("run_id"),
+                "metrics": [MetricDto.from_dict(_item) for _item in obj.get("metrics")]
+                if obj.get("metrics") is not None
+                else None,
+                "params": [ParamDto.from_dict(_item) for _item in obj.get("params")]
+                if obj.get("params") is not None
+                else None,
+                "tags": [RunTagDto.from_dict(_item) for _item in obj.get("tags")]
+                if obj.get("tags") is not None
+                else None,
+            }
+        )
+        return _obj

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

@@ -0,0 +1,80 @@
+# 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, Union
+
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictFloat, StrictInt, StrictStr
+
+
+class LogMetricRequestDto(BaseModel):
+    """
+    LogMetricRequestDto
+    """
+
+    run_id: Optional[StrictStr] = None
+    run_uuid: Optional[StrictStr] = None
+    key: StrictStr = Field(...)
+    value: Union[StrictFloat, StrictInt] = Field(...)
+    timestamp: StrictInt = Field(...)
+    step: Optional[StrictInt] = 0
+    __properties = ["run_id", "run_uuid", "key", "value", "timestamp", "step"]
+
+    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) -> LogMetricRequestDto:
+        """Create an instance of LogMetricRequestDto 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) -> LogMetricRequestDto:
+        """Create an instance of LogMetricRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return LogMetricRequestDto.parse_obj(obj)
+
+        _obj = LogMetricRequestDto.parse_obj(
+            {
+                "run_id": obj.get("run_id"),
+                "run_uuid": obj.get("run_uuid"),
+                "key": obj.get("key"),
+                "value": obj.get("value"),
+                "timestamp": obj.get("timestamp"),
+                "step": obj.get("step") if obj.get("step") is not None else 0,
+            }
+        )
+        return _obj

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

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

@@ -0,0 +1,37 @@
+# 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 Method(str, Enum):
+    """
+    +sort=50 +uiType=Hidden
+    """
+
+    """
+    allowed enum values
+    """
+    GET = "get"
+    POST = "post"
+    PUT = "put"
+    DELETE = "delete"
+    PATCH = "patch"
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Method:
+        """Create an instance of Method from a JSON string"""
+        return Method(json.loads(json_str))

truefoundry/ml/autogen/client/models/metric_collection_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.metric_dto import MetricDto
+from truefoundry.pydantic_v1 import BaseModel, StrictStr, conlist
+
+
+class MetricCollectionDto(BaseModel):
+    """
+    MetricCollectionDto
+    """
+
+    key: Optional[StrictStr] = None
+    metrics: Optional[conlist(MetricDto)] = None
+    __properties = ["key", "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) -> MetricCollectionDto:
+        """Create an instance of MetricCollectionDto 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) -> MetricCollectionDto:
+        """Create an instance of MetricCollectionDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return MetricCollectionDto.parse_obj(obj)
+
+        _obj = MetricCollectionDto.parse_obj(
+            {
+                "key": obj.get("key"),
+                "metrics": [MetricDto.from_dict(_item) for _item in obj.get("metrics")]
+                if obj.get("metrics") is not None
+                else None,
+            }
+        )
+        return _obj