orca-sdk 0.0.93__py3-none-any.whl → 0.0.95__py3-none-any.whl

orca_sdk/_generated_api_client/models/regression_evaluation_request.py

@@ -0,0 +1,148 @@
+"""
+This file is generated by the openapi-python-client tool via the generate_api_client.py script
+
+It is a customized template from the openapi-python-client tool's default template:
+https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
+
+The main change is:
+- Fix typing issues
+"""
+
+# flake8: noqa: C901
+
+from typing import Any, List, Type, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="RegressionEvaluationRequest")
+
+
+@_attrs_define
+class RegressionEvaluationRequest:
+    """
+    Attributes:
+        datasource_id (str):
+        datasource_score_column (str):
+        datasource_value_column (str):
+        memoryset_override_id (Union[None, Unset, str]):
+        record_telemetry (Union[Unset, bool]):  Default: False.
+        telemetry_tags (Union[List[str], None, Unset]):
+    """
+
+    datasource_id: str
+    datasource_score_column: str
+    datasource_value_column: str
+    memoryset_override_id: Union[None, Unset, str] = UNSET
+    record_telemetry: Union[Unset, bool] = False
+    telemetry_tags: Union[List[str], None, Unset] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        datasource_id = self.datasource_id
+
+        datasource_score_column = self.datasource_score_column
+
+        datasource_value_column = self.datasource_value_column
+
+        memoryset_override_id: Union[None, Unset, str]
+        if isinstance(self.memoryset_override_id, Unset):
+            memoryset_override_id = UNSET
+        else:
+            memoryset_override_id = self.memoryset_override_id
+
+        record_telemetry = self.record_telemetry
+
+        telemetry_tags: Union[List[str], None, Unset]
+        if isinstance(self.telemetry_tags, Unset):
+            telemetry_tags = UNSET
+        elif isinstance(self.telemetry_tags, list):
+            telemetry_tags = self.telemetry_tags
+
+        else:
+            telemetry_tags = self.telemetry_tags
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "datasource_id": datasource_id,
+                "datasource_score_column": datasource_score_column,
+                "datasource_value_column": datasource_value_column,
+            }
+        )
+        if memoryset_override_id is not UNSET:
+            field_dict["memoryset_override_id"] = memoryset_override_id
+        if record_telemetry is not UNSET:
+            field_dict["record_telemetry"] = record_telemetry
+        if telemetry_tags is not UNSET:
+            field_dict["telemetry_tags"] = telemetry_tags
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        datasource_id = d.pop("datasource_id")
+
+        datasource_score_column = d.pop("datasource_score_column")
+
+        datasource_value_column = d.pop("datasource_value_column")
+
+        def _parse_memoryset_override_id(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        memoryset_override_id = _parse_memoryset_override_id(d.pop("memoryset_override_id", UNSET))
+
+        record_telemetry = d.pop("record_telemetry", UNSET)
+
+        def _parse_telemetry_tags(data: object) -> Union[List[str], None, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                telemetry_tags_type_0 = cast(List[str], data)
+
+                return telemetry_tags_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[List[str], None, Unset], data)
+
+        telemetry_tags = _parse_telemetry_tags(d.pop("telemetry_tags", UNSET))
+
+        regression_evaluation_request = cls(
+            datasource_id=datasource_id,
+            datasource_score_column=datasource_score_column,
+            datasource_value_column=datasource_value_column,
+            memoryset_override_id=memoryset_override_id,
+            record_telemetry=record_telemetry,
+            telemetry_tags=telemetry_tags,
+        )
+
+        regression_evaluation_request.additional_properties = d
+        return regression_evaluation_request
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

orca_sdk/_generated_api_client/models/regression_metrics.py

@@ -0,0 +1,172 @@
+"""
+This file is generated by the openapi-python-client tool via the generate_api_client.py script
+
+It is a customized template from the openapi-python-client tool's default template:
+https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
+
+The main change is:
+- Fix typing issues
+"""
+
+# flake8: noqa: C901
+
+from typing import Any, Type, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="RegressionMetrics")
+
+
+@_attrs_define
+class RegressionMetrics:
+    """
+    Attributes:
+        mse (float):
+        rmse (float):
+        mae (float):
+        r2 (float):
+        explained_variance (float):
+        loss (float):
+        anomaly_score_mean (Union[None, Unset, float]):
+        anomaly_score_median (Union[None, Unset, float]):
+        anomaly_score_variance (Union[None, Unset, float]):
+    """
+
+    mse: float
+    rmse: float
+    mae: float
+    r2: float
+    explained_variance: float
+    loss: float
+    anomaly_score_mean: Union[None, Unset, float] = UNSET
+    anomaly_score_median: Union[None, Unset, float] = UNSET
+    anomaly_score_variance: Union[None, Unset, float] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        mse = self.mse
+
+        rmse = self.rmse
+
+        mae = self.mae
+
+        r2 = self.r2
+
+        explained_variance = self.explained_variance
+
+        loss = self.loss
+
+        anomaly_score_mean: Union[None, Unset, float]
+        if isinstance(self.anomaly_score_mean, Unset):
+            anomaly_score_mean = UNSET
+        else:
+            anomaly_score_mean = self.anomaly_score_mean
+
+        anomaly_score_median: Union[None, Unset, float]
+        if isinstance(self.anomaly_score_median, Unset):
+            anomaly_score_median = UNSET
+        else:
+            anomaly_score_median = self.anomaly_score_median
+
+        anomaly_score_variance: Union[None, Unset, float]
+        if isinstance(self.anomaly_score_variance, Unset):
+            anomaly_score_variance = UNSET
+        else:
+            anomaly_score_variance = self.anomaly_score_variance
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "mse": mse,
+                "rmse": rmse,
+                "mae": mae,
+                "r2": r2,
+                "explained_variance": explained_variance,
+                "loss": loss,
+            }
+        )
+        if anomaly_score_mean is not UNSET:
+            field_dict["anomaly_score_mean"] = anomaly_score_mean
+        if anomaly_score_median is not UNSET:
+            field_dict["anomaly_score_median"] = anomaly_score_median
+        if anomaly_score_variance is not UNSET:
+            field_dict["anomaly_score_variance"] = anomaly_score_variance
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        mse = d.pop("mse")
+
+        rmse = d.pop("rmse")
+
+        mae = d.pop("mae")
+
+        r2 = d.pop("r2")
+
+        explained_variance = d.pop("explained_variance")
+
+        loss = d.pop("loss")
+
+        def _parse_anomaly_score_mean(data: object) -> Union[None, Unset, float]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, float], data)
+
+        anomaly_score_mean = _parse_anomaly_score_mean(d.pop("anomaly_score_mean", UNSET))
+
+        def _parse_anomaly_score_median(data: object) -> Union[None, Unset, float]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, float], data)
+
+        anomaly_score_median = _parse_anomaly_score_median(d.pop("anomaly_score_median", UNSET))
+
+        def _parse_anomaly_score_variance(data: object) -> Union[None, Unset, float]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, float], data)
+
+        anomaly_score_variance = _parse_anomaly_score_variance(d.pop("anomaly_score_variance", UNSET))
+
+        regression_metrics = cls(
+            mse=mse,
+            rmse=rmse,
+            mae=mae,
+            r2=r2,
+            explained_variance=explained_variance,
+            loss=loss,
+            anomaly_score_mean=anomaly_score_mean,
+            anomaly_score_median=anomaly_score_median,
+            anomaly_score_variance=anomaly_score_variance,
+        )
+
+        regression_metrics.additional_properties = d
+        return regression_metrics
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

orca_sdk/_generated_api_client/models/regression_model_metadata.py

@@ -0,0 +1,177 @@
+"""
+This file is generated by the openapi-python-client tool via the generate_api_client.py script
+
+It is a customized template from the openapi-python-client tool's default template:
+https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
+
+The main change is:
+- Fix typing issues
+"""
+
+# flake8: noqa: C901
+
+import datetime
+from enum import Enum
+from typing import Any, Type, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+from dateutil.parser import isoparse
+
+from ..models.rar_head_type import RARHeadType
+
+T = TypeVar("T", bound="RegressionModelMetadata")
+
+
+@_attrs_define
+class RegressionModelMetadata:
+    """
+    Attributes:
+        id (str):
+        org_id (str):
+        name (str):
+        description (Union[None, str]):
+        version (int):
+        memoryset_id (str):
+        memory_lookup_count (int):
+        storage_path (str):
+        memoryset_collection_name (str):
+        created_at (datetime.datetime):
+        updated_at (datetime.datetime):
+        locked (bool):
+        head_type (RARHeadType):
+    """
+
+    id: str
+    org_id: str
+    name: str
+    description: Union[None, str]
+    version: int
+    memoryset_id: str
+    memory_lookup_count: int
+    storage_path: str
+    memoryset_collection_name: str
+    created_at: datetime.datetime
+    updated_at: datetime.datetime
+    locked: bool
+    head_type: RARHeadType
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        id = self.id
+
+        org_id = self.org_id
+
+        name = self.name
+
+        description: Union[None, str]
+        description = self.description
+
+        version = self.version
+
+        memoryset_id = self.memoryset_id
+
+        memory_lookup_count = self.memory_lookup_count
+
+        storage_path = self.storage_path
+
+        memoryset_collection_name = self.memoryset_collection_name
+
+        created_at = self.created_at.isoformat()
+
+        updated_at = self.updated_at.isoformat()
+
+        locked = self.locked
+
+        head_type = self.head_type.value if isinstance(self.head_type, Enum) else self.head_type
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "id": id,
+                "org_id": org_id,
+                "name": name,
+                "description": description,
+                "version": version,
+                "memoryset_id": memoryset_id,
+                "memory_lookup_count": memory_lookup_count,
+                "storage_path": storage_path,
+                "memoryset_collection_name": memoryset_collection_name,
+                "created_at": created_at,
+                "updated_at": updated_at,
+                "locked": locked,
+                "head_type": head_type,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        id = d.pop("id")
+
+        org_id = d.pop("org_id")
+
+        name = d.pop("name")
+
+        def _parse_description(data: object) -> Union[None, str]:
+            if data is None:
+                return data
+            return cast(Union[None, str], data)
+
+        description = _parse_description(d.pop("description"))
+
+        version = d.pop("version")
+
+        memoryset_id = d.pop("memoryset_id")
+
+        memory_lookup_count = d.pop("memory_lookup_count")
+
+        storage_path = d.pop("storage_path")
+
+        memoryset_collection_name = d.pop("memoryset_collection_name")
+
+        created_at = isoparse(d.pop("created_at"))
+
+        updated_at = isoparse(d.pop("updated_at"))
+
+        locked = d.pop("locked")
+
+        head_type = RARHeadType(d.pop("head_type"))
+
+        regression_model_metadata = cls(
+            id=id,
+            org_id=org_id,
+            name=name,
+            description=description,
+            version=version,
+            memoryset_id=memoryset_id,
+            memory_lookup_count=memory_lookup_count,
+            storage_path=storage_path,
+            memoryset_collection_name=memoryset_collection_name,
+            created_at=created_at,
+            updated_at=updated_at,
+            locked=locked,
+            head_type=head_type,
+        )
+
+        regression_model_metadata.additional_properties = d
+        return regression_model_metadata
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

orca_sdk/_generated_api_client/models/regression_prediction_request.py

@@ -0,0 +1,195 @@
+"""
+This file is generated by the openapi-python-client tool via the generate_api_client.py script
+
+It is a customized template from the openapi-python-client tool's default template:
+https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
+
+The main change is:
+- Fix typing issues
+"""
+
+# flake8: noqa: C901
+
+from typing import Any, List, Type, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="RegressionPredictionRequest")
+
+
+@_attrs_define
+class RegressionPredictionRequest:
+    """
+    Attributes:
+        input_values (List[str]):
+        expected_scores (Union[List[float], None, Unset]):
+        tags (Union[Unset, List[str]]):
+        memoryset_override_id (Union[None, Unset, str]):
+        save_telemetry (Union[Unset, bool]):  Default: True.
+        save_telemetry_synchronously (Union[Unset, bool]):  Default: False.
+    """
+
+    input_values: List[str]
+    expected_scores: Union[List[float], None, Unset] = UNSET
+    tags: Union[Unset, List[str]] = UNSET
+    memoryset_override_id: Union[None, Unset, str] = UNSET
+    save_telemetry: Union[Unset, bool] = True
+    save_telemetry_synchronously: Union[Unset, bool] = False
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        input_values: List[str]
+        if isinstance(self.input_values, list):
+            input_values = self.input_values
+
+        expected_scores: Union[List[float], None, Unset]
+        if isinstance(self.expected_scores, Unset):
+            expected_scores = UNSET
+        elif isinstance(self.expected_scores, list):
+            expected_scores = self.expected_scores
+
+        else:
+            expected_scores = self.expected_scores
+
+        tags: Union[Unset, List[str]] = UNSET
+        if not isinstance(self.tags, Unset):
+            tags = self.tags
+
+        memoryset_override_id: Union[None, Unset, str]
+        if isinstance(self.memoryset_override_id, Unset):
+            memoryset_override_id = UNSET
+        else:
+            memoryset_override_id = self.memoryset_override_id
+
+        save_telemetry = self.save_telemetry
+
+        save_telemetry_synchronously = self.save_telemetry_synchronously
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "input_values": input_values,
+            }
+        )
+        if expected_scores is not UNSET:
+            field_dict["expected_scores"] = expected_scores
+        if tags is not UNSET:
+            field_dict["tags"] = tags
+        if memoryset_override_id is not UNSET:
+            field_dict["memoryset_override_id"] = memoryset_override_id
+        if save_telemetry is not UNSET:
+            field_dict["save_telemetry"] = save_telemetry
+        if save_telemetry_synchronously is not UNSET:
+            field_dict["save_telemetry_synchronously"] = save_telemetry_synchronously
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+
+        def _parse_input_values(data: object) -> List[str]:
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                input_values_type_0 = cast(List[str], data)
+
+                return input_values_type_0
+            except:  # noqa: E722
+                pass
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                input_values_type_1 = cast(List[str], data)
+
+                return input_values_type_1
+            except:  # noqa: E722
+                pass
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                input_values_type_2 = cast(List[str], data)
+
+                return input_values_type_2
+            except:  # noqa: E722
+                pass
+            if not isinstance(data, list):
+                raise TypeError()
+            input_values_type_3 = []
+            _input_values_type_3 = data
+            for input_values_type_3_item_data in _input_values_type_3:
+
+                def _parse_input_values_type_3_item(data: object) -> str:
+                    return cast(str, data)
+
+                input_values_type_3_item = _parse_input_values_type_3_item(input_values_type_3_item_data)
+
+                input_values_type_3.append(input_values_type_3_item)
+
+            return input_values_type_3
+
+        input_values = _parse_input_values(d.pop("input_values"))
+
+        def _parse_expected_scores(data: object) -> Union[List[float], None, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                expected_scores_type_0 = cast(List[float], data)
+
+                return expected_scores_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[List[float], None, Unset], data)
+
+        expected_scores = _parse_expected_scores(d.pop("expected_scores", UNSET))
+
+        tags = cast(List[str], d.pop("tags", UNSET))
+
+        def _parse_memoryset_override_id(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        memoryset_override_id = _parse_memoryset_override_id(d.pop("memoryset_override_id", UNSET))
+
+        save_telemetry = d.pop("save_telemetry", UNSET)
+
+        save_telemetry_synchronously = d.pop("save_telemetry_synchronously", UNSET)
+
+        regression_prediction_request = cls(
+            input_values=input_values,
+            expected_scores=expected_scores,
+            tags=tags,
+            memoryset_override_id=memoryset_override_id,
+            save_telemetry=save_telemetry,
+            save_telemetry_synchronously=save_telemetry_synchronously,
+        )
+
+        regression_prediction_request.additional_properties = d
+        return regression_prediction_request
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties