robodock-cli 0.1.0__py3-none-any.whl

fast_api_client/models/run_create_request.py

@@ -0,0 +1,243 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.run_create_request_configs import RunCreateRequestConfigs
+
+
+T = TypeVar("T", bound="RunCreateRequest")
+
+
+@_attrs_define
+class RunCreateRequest:
+    """
+    Attributes:
+        creator (str): 创建人
+        project_name (str): 项目名
+        run_name (str): 任务名
+        train_arc (str): 模型架构: Gr00t N1.6/Fii_FALCON_0/FII_FALCON_SMT
+        configs (RunCreateRequestConfigs): Run configurations
+        description (None | str | Unset): 任务描述
+        label (None | str | Unset): 标签
+        robot_type (None | str | Unset): 机器人类型: fii_w2/actibot/umi
+        check_points_path (list[str] | None | Unset): check_points 文件地址
+        check_points_onnx_path (list[str] | None | Unset): onnx/edge checkpoint 文件地址
+        extra_info (None | str | Unset): 额外信息
+    """
+
+    creator: str
+    project_name: str
+    run_name: str
+    train_arc: str
+    configs: RunCreateRequestConfigs
+    description: None | str | Unset = UNSET
+    label: None | str | Unset = UNSET
+    robot_type: None | str | Unset = UNSET
+    check_points_path: list[str] | None | Unset = UNSET
+    check_points_onnx_path: list[str] | None | Unset = UNSET
+    extra_info: None | str | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        creator = self.creator
+
+        project_name = self.project_name
+
+        run_name = self.run_name
+
+        train_arc = self.train_arc
+
+        configs = self.configs.to_dict()
+
+        description: None | str | Unset
+        if isinstance(self.description, Unset):
+            description = UNSET
+        else:
+            description = self.description
+
+        label: None | str | Unset
+        if isinstance(self.label, Unset):
+            label = UNSET
+        else:
+            label = self.label
+
+        robot_type: None | str | Unset
+        if isinstance(self.robot_type, Unset):
+            robot_type = UNSET
+        else:
+            robot_type = self.robot_type
+
+        check_points_path: list[str] | None | Unset
+        if isinstance(self.check_points_path, Unset):
+            check_points_path = UNSET
+        elif isinstance(self.check_points_path, list):
+            check_points_path = self.check_points_path
+
+        else:
+            check_points_path = self.check_points_path
+
+        check_points_onnx_path: list[str] | None | Unset
+        if isinstance(self.check_points_onnx_path, Unset):
+            check_points_onnx_path = UNSET
+        elif isinstance(self.check_points_onnx_path, list):
+            check_points_onnx_path = self.check_points_onnx_path
+
+        else:
+            check_points_onnx_path = self.check_points_onnx_path
+
+        extra_info: None | str | Unset
+        if isinstance(self.extra_info, Unset):
+            extra_info = UNSET
+        else:
+            extra_info = self.extra_info
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "creator": creator,
+                "project_name": project_name,
+                "run_name": run_name,
+                "train_arc": train_arc,
+                "configs": configs,
+            }
+        )
+        if description is not UNSET:
+            field_dict["description"] = description
+        if label is not UNSET:
+            field_dict["label"] = label
+        if robot_type is not UNSET:
+            field_dict["robot_type"] = robot_type
+        if check_points_path is not UNSET:
+            field_dict["check_points_path"] = check_points_path
+        if check_points_onnx_path is not UNSET:
+            field_dict["check_points_onnx_path"] = check_points_onnx_path
+        if extra_info is not UNSET:
+            field_dict["extra_info"] = extra_info
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.run_create_request_configs import RunCreateRequestConfigs  # noqa: PLC0415
+
+        d = dict(src_dict)
+        creator = d.pop("creator")
+
+        project_name = d.pop("project_name")
+
+        run_name = d.pop("run_name")
+
+        train_arc = d.pop("train_arc")
+
+        configs = RunCreateRequestConfigs.from_dict(d.pop("configs"))
+
+        def _parse_description(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        description = _parse_description(d.pop("description", UNSET))
+
+        def _parse_label(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        label = _parse_label(d.pop("label", UNSET))
+
+        def _parse_robot_type(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        robot_type = _parse_robot_type(d.pop("robot_type", UNSET))
+
+        def _parse_check_points_path(data: object) -> list[str] | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                check_points_path_type_0 = cast(list[str], data)
+
+                return check_points_path_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(list[str] | None | Unset, data)
+
+        check_points_path = _parse_check_points_path(d.pop("check_points_path", UNSET))
+
+        def _parse_check_points_onnx_path(data: object) -> list[str] | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                check_points_onnx_path_type_0 = cast(list[str], data)
+
+                return check_points_onnx_path_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(list[str] | None | Unset, data)
+
+        check_points_onnx_path = _parse_check_points_onnx_path(d.pop("check_points_onnx_path", UNSET))
+
+        def _parse_extra_info(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        extra_info = _parse_extra_info(d.pop("extra_info", UNSET))
+
+        run_create_request = cls(
+            creator=creator,
+            project_name=project_name,
+            run_name=run_name,
+            train_arc=train_arc,
+            configs=configs,
+            description=description,
+            label=label,
+            robot_type=robot_type,
+            check_points_path=check_points_path,
+            check_points_onnx_path=check_points_onnx_path,
+            extra_info=extra_info,
+        )
+
+        run_create_request.additional_properties = d
+        return run_create_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

fast_api_client/models/run_create_request_configs.py

@@ -0,0 +1,47 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="RunCreateRequestConfigs")
+
+
+@_attrs_define
+class RunCreateRequestConfigs:
+    """Run configurations"""
+
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        run_create_request_configs = cls()
+
+        run_create_request_configs.additional_properties = d
+        return run_create_request_configs
+
+    @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