anyscale 0.26.43__py3-none-any.whl → 0.26.44__py3-none-any.whl

anyscale/client/openapi_client/models/workload_state_info.py

@@ -0,0 +1,295 @@
+# coding: utf-8
+
+"""
+    Managed Ray API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)  # noqa: E501
+
+    The version of the OpenAPI document: 0.1.0
+    Generated by: https://openapi-generator.tech
+"""
+
+
+import pprint
+import re  # noqa: F401
+
+import six
+
+from openapi_client.configuration import Configuration
+
+
+class WorkloadStateInfo(object):
+    """NOTE: This class is auto generated by OpenAPI Generator.
+    Ref: https://openapi-generator.tech
+
+    Do not edit the class manually.
+    """
+
+    """
+    Attributes:
+      openapi_types (dict): The key is attribute name
+                            and the value is attribute type.
+      attribute_map (dict): The key is attribute name
+                            and the value is json key in definition.
+    """
+    openapi_types = {
+        'workload_name': 'str',
+        'workload_type': 'str',
+        'workload_start_time': 'datetime',
+        'workload_cloud': 'str',
+        'workload_project': 'str',
+        'workload_creator': 'str',
+        'workload_machine_info': 'list[WorkloadMachineInfo]'
+    }
+
+    attribute_map = {
+        'workload_name': 'workload_name',
+        'workload_type': 'workload_type',
+        'workload_start_time': 'workload_start_time',
+        'workload_cloud': 'workload_cloud',
+        'workload_project': 'workload_project',
+        'workload_creator': 'workload_creator',
+        'workload_machine_info': 'workload_machine_info'
+    }
+
+    def __init__(self, workload_name=None, workload_type=None, workload_start_time=None, workload_cloud=None, workload_project='', workload_creator='', workload_machine_info=None, local_vars_configuration=None):  # noqa: E501
+        """WorkloadStateInfo - a model defined in OpenAPI"""  # noqa: E501
+        if local_vars_configuration is None:
+            local_vars_configuration = Configuration()
+        self.local_vars_configuration = local_vars_configuration
+
+        self._workload_name = None
+        self._workload_type = None
+        self._workload_start_time = None
+        self._workload_cloud = None
+        self._workload_project = None
+        self._workload_creator = None
+        self._workload_machine_info = None
+        self.discriminator = None
+
+        self.workload_name = workload_name
+        self.workload_type = workload_type
+        self.workload_start_time = workload_start_time
+        self.workload_cloud = workload_cloud
+        if workload_project is not None:
+            self.workload_project = workload_project
+        if workload_creator is not None:
+            self.workload_creator = workload_creator
+        self.workload_machine_info = workload_machine_info
+
+    @property
+    def workload_name(self):
+        """Gets the workload_name of this WorkloadStateInfo.  # noqa: E501
+
+        The name of the workload.  # noqa: E501
+
+        :return: The workload_name of this WorkloadStateInfo.  # noqa: E501
+        :rtype: str
+        """
+        return self._workload_name
+
+    @workload_name.setter
+    def workload_name(self, workload_name):
+        """Sets the workload_name of this WorkloadStateInfo.
+
+        The name of the workload.  # noqa: E501
+
+        :param workload_name: The workload_name of this WorkloadStateInfo.  # noqa: E501
+        :type: str
+        """
+        if self.local_vars_configuration.client_side_validation and workload_name is None:  # noqa: E501
+            raise ValueError("Invalid value for `workload_name`, must not be `None`")  # noqa: E501
+
+        self._workload_name = workload_name
+
+    @property
+    def workload_type(self):
+        """Gets the workload_type of this WorkloadStateInfo.  # noqa: E501
+
+        The type of the workload.  # noqa: E501
+
+        :return: The workload_type of this WorkloadStateInfo.  # noqa: E501
+        :rtype: str
+        """
+        return self._workload_type
+
+    @workload_type.setter
+    def workload_type(self, workload_type):
+        """Sets the workload_type of this WorkloadStateInfo.
+
+        The type of the workload.  # noqa: E501
+
+        :param workload_type: The workload_type of this WorkloadStateInfo.  # noqa: E501
+        :type: str
+        """
+        if self.local_vars_configuration.client_side_validation and workload_type is None:  # noqa: E501
+            raise ValueError("Invalid value for `workload_type`, must not be `None`")  # noqa: E501
+
+        self._workload_type = workload_type
+
+    @property
+    def workload_start_time(self):
+        """Gets the workload_start_time of this WorkloadStateInfo.  # noqa: E501
+
+        The start time of the workload.  # noqa: E501
+
+        :return: The workload_start_time of this WorkloadStateInfo.  # noqa: E501
+        :rtype: datetime
+        """
+        return self._workload_start_time
+
+    @workload_start_time.setter
+    def workload_start_time(self, workload_start_time):
+        """Sets the workload_start_time of this WorkloadStateInfo.
+
+        The start time of the workload.  # noqa: E501
+
+        :param workload_start_time: The workload_start_time of this WorkloadStateInfo.  # noqa: E501
+        :type: datetime
+        """
+        if self.local_vars_configuration.client_side_validation and workload_start_time is None:  # noqa: E501
+            raise ValueError("Invalid value for `workload_start_time`, must not be `None`")  # noqa: E501
+
+        self._workload_start_time = workload_start_time
+
+    @property
+    def workload_cloud(self):
+        """Gets the workload_cloud of this WorkloadStateInfo.  # noqa: E501
+
+        The cloud of the workload.  # noqa: E501
+
+        :return: The workload_cloud of this WorkloadStateInfo.  # noqa: E501
+        :rtype: str
+        """
+        return self._workload_cloud
+
+    @workload_cloud.setter
+    def workload_cloud(self, workload_cloud):
+        """Sets the workload_cloud of this WorkloadStateInfo.
+
+        The cloud of the workload.  # noqa: E501
+
+        :param workload_cloud: The workload_cloud of this WorkloadStateInfo.  # noqa: E501
+        :type: str
+        """
+        if self.local_vars_configuration.client_side_validation and workload_cloud is None:  # noqa: E501
+            raise ValueError("Invalid value for `workload_cloud`, must not be `None`")  # noqa: E501
+
+        self._workload_cloud = workload_cloud
+
+    @property
+    def workload_project(self):
+        """Gets the workload_project of this WorkloadStateInfo.  # noqa: E501
+
+        The project of the workload.  # noqa: E501
+
+        :return: The workload_project of this WorkloadStateInfo.  # noqa: E501
+        :rtype: str
+        """
+        return self._workload_project
+
+    @workload_project.setter
+    def workload_project(self, workload_project):
+        """Sets the workload_project of this WorkloadStateInfo.
+
+        The project of the workload.  # noqa: E501
+
+        :param workload_project: The workload_project of this WorkloadStateInfo.  # noqa: E501
+        :type: str
+        """
+
+        self._workload_project = workload_project
+
+    @property
+    def workload_creator(self):
+        """Gets the workload_creator of this WorkloadStateInfo.  # noqa: E501
+
+        The creator of the workload.  # noqa: E501
+
+        :return: The workload_creator of this WorkloadStateInfo.  # noqa: E501
+        :rtype: str
+        """
+        return self._workload_creator
+
+    @workload_creator.setter
+    def workload_creator(self, workload_creator):
+        """Sets the workload_creator of this WorkloadStateInfo.
+
+        The creator of the workload.  # noqa: E501
+
+        :param workload_creator: The workload_creator of this WorkloadStateInfo.  # noqa: E501
+        :type: str
+        """
+
+        self._workload_creator = workload_creator
+
+    @property
+    def workload_machine_info(self):
+        """Gets the workload_machine_info of this WorkloadStateInfo.  # noqa: E501
+
+        The machines allocated to this workload.  # noqa: E501
+
+        :return: The workload_machine_info of this WorkloadStateInfo.  # noqa: E501
+        :rtype: list[WorkloadMachineInfo]
+        """
+        return self._workload_machine_info
+
+    @workload_machine_info.setter
+    def workload_machine_info(self, workload_machine_info):
+        """Sets the workload_machine_info of this WorkloadStateInfo.
+
+        The machines allocated to this workload.  # noqa: E501
+
+        :param workload_machine_info: The workload_machine_info of this WorkloadStateInfo.  # noqa: E501
+        :type: list[WorkloadMachineInfo]
+        """
+        if self.local_vars_configuration.client_side_validation and workload_machine_info is None:  # noqa: E501
+            raise ValueError("Invalid value for `workload_machine_info`, must not be `None`")  # noqa: E501
+
+        self._workload_machine_info = workload_machine_info
+
+    def to_dict(self):
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.openapi_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = value
+
+        return result
+
+    def to_str(self):
+        """Returns the string representation of the model"""
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self):
+        """For `print` and `pprint`"""
+        return self.to_str()
+
+    def __eq__(self, other):
+        """Returns true if both objects are equal"""
+        if not isinstance(other, WorkloadStateInfo):
+            return False
+
+        return self.to_dict() == other.to_dict()
+
+    def __ne__(self, other):
+        """Returns true if both objects are not equal"""
+        if not isinstance(other, WorkloadStateInfo):
+            return True
+
+        return self.to_dict() != other.to_dict()

anyscale/client/openapi_client/models/workloadstateinfo_list_response.py

@@ -0,0 +1,147 @@
+# coding: utf-8
+
+"""
+    Managed Ray API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)  # noqa: E501
+
+    The version of the OpenAPI document: 0.1.0
+    Generated by: https://openapi-generator.tech
+"""
+
+
+import pprint
+import re  # noqa: F401
+
+import six
+
+from openapi_client.configuration import Configuration
+
+
+class WorkloadstateinfoListResponse(object):
+    """NOTE: This class is auto generated by OpenAPI Generator.
+    Ref: https://openapi-generator.tech
+
+    Do not edit the class manually.
+    """
+
+    """
+    Attributes:
+      openapi_types (dict): The key is attribute name
+                            and the value is attribute type.
+      attribute_map (dict): The key is attribute name
+                            and the value is json key in definition.
+    """
+    openapi_types = {
+        'results': 'list[WorkloadStateInfo]',
+        'metadata': 'ListResponseMetadata'
+    }
+
+    attribute_map = {
+        'results': 'results',
+        'metadata': 'metadata'
+    }
+
+    def __init__(self, results=None, metadata=None, local_vars_configuration=None):  # noqa: E501
+        """WorkloadstateinfoListResponse - a model defined in OpenAPI"""  # noqa: E501
+        if local_vars_configuration is None:
+            local_vars_configuration = Configuration()
+        self.local_vars_configuration = local_vars_configuration
+
+        self._results = None
+        self._metadata = None
+        self.discriminator = None
+
+        self.results = results
+        if metadata is not None:
+            self.metadata = metadata
+
+    @property
+    def results(self):
+        """Gets the results of this WorkloadstateinfoListResponse.  # noqa: E501
+
+
+        :return: The results of this WorkloadstateinfoListResponse.  # noqa: E501
+        :rtype: list[WorkloadStateInfo]
+        """
+        return self._results
+
+    @results.setter
+    def results(self, results):
+        """Sets the results of this WorkloadstateinfoListResponse.
+
+
+        :param results: The results of this WorkloadstateinfoListResponse.  # noqa: E501
+        :type: list[WorkloadStateInfo]
+        """
+        if self.local_vars_configuration.client_side_validation and results is None:  # noqa: E501
+            raise ValueError("Invalid value for `results`, must not be `None`")  # noqa: E501
+
+        self._results = results
+
+    @property
+    def metadata(self):
+        """Gets the metadata of this WorkloadstateinfoListResponse.  # noqa: E501
+
+
+        :return: The metadata of this WorkloadstateinfoListResponse.  # noqa: E501
+        :rtype: ListResponseMetadata
+        """
+        return self._metadata
+
+    @metadata.setter
+    def metadata(self, metadata):
+        """Sets the metadata of this WorkloadstateinfoListResponse.
+
+
+        :param metadata: The metadata of this WorkloadstateinfoListResponse.  # noqa: E501
+        :type: ListResponseMetadata
+        """
+
+        self._metadata = metadata
+
+    def to_dict(self):
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.openapi_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = value
+
+        return result
+
+    def to_str(self):
+        """Returns the string representation of the model"""
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self):
+        """For `print` and `pprint`"""
+        return self.to_str()
+
+    def __eq__(self, other):
+        """Returns true if both objects are equal"""
+        if not isinstance(other, WorkloadstateinfoListResponse):
+            return False
+
+        return self.to_dict() == other.to_dict()
+
+    def __ne__(self, other):
+        """Returns true if both objects are not equal"""
+        if not isinstance(other, WorkloadstateinfoListResponse):
+            return True
+
+        return self.to_dict() != other.to_dict()

anyscale/commands/cloud_commands.py

@@ -926,6 +926,7 @@ def register_cloud(  # noqa: PLR0913, PLR0912, C901
             nfs_mount_targets=list(nfs_mount_target) if nfs_mount_target else [],
             nfs_mount_path=nfs_mount_path,
             kubernetes_zones=kubernetes_zones.split(",") if kubernetes_zones else [],
+            anyscale_operator_iam_identity=anyscale_operator_iam_identity,
         )
 
     else:

anyscale/commands/compute_config_commands.py

@@ -8,7 +8,7 @@ import anyscale
 from anyscale.cli_logger import BlockLogger
 from anyscale.commands import command_examples
 from anyscale.commands.util import AnyscaleCommand, LegacyAnyscaleCommand
-from anyscale.compute_config import ComputeConfig
+from anyscale.compute_config import ComputeConfig, MultiDeploymentComputeConfig
 from anyscale.controllers.compute_config_controller import ComputeConfigController
 from anyscale.util import validate_non_negative_arg
 
@@ -105,8 +105,15 @@ def create_compute_config(
     if compute_config_file is not None:
         ComputeConfigController().create(compute_config_file, name)
     elif config_file is not None:
-        config = ComputeConfig.from_yaml(config_file)
-        anyscale.compute_config.create(config, name=name)
+        try:
+            config = ComputeConfig.from_yaml(config_file)
+        except TypeError:
+            config = MultiDeploymentComputeConfig.from_yaml(config_file)
+
+        if isinstance(config, ComputeConfig):
+            anyscale.compute_config.create(config, name=name)
+        elif isinstance(config, MultiDeploymentComputeConfig):
+            anyscale.compute_config.create_multi_deployment(config, name=name)
     else:
         raise click.ClickException(
             "Either the --config-file flag or [COMPUTE_CONFIG_FILE] argument must be provided."

anyscale/compute_config/__init__.py

@@ -10,16 +10,20 @@ from anyscale.compute_config.commands import (
     _ARCHIVE_EXAMPLE,
     _CREATE_ARG_DOCSTRINGS,
     _CREATE_EXAMPLE,
+    _CREATE_MULTI_DEPLOYMENT_ARG_DOCSTRINGS,
+    _CREATE_MULTI_DEPLOYMENT_EXAMPLE,
     _GET_ARG_DOCSTRINGS,
     _GET_EXAMPLE,
     archive,
     create,
+    create_multi_deployment,
     get,
 )
 from anyscale.compute_config.models import (
     ComputeConfig,
     ComputeConfigVersion,
     HeadNodeConfig,
+    MultiDeploymentComputeConfig,
     WorkerNodeGroupConfig,
 )
 
@@ -49,6 +53,18 @@ class ComputeConfigSDK:
         full_name, _ = self._private_sdk.create_compute_config(config, name=name)
         return full_name
 
+    @sdk_docs(
+        doc_py_example=_CREATE_MULTI_DEPLOYMENT_EXAMPLE,
+        arg_docstrings=_CREATE_MULTI_DEPLOYMENT_ARG_DOCSTRINGS,
+    )
+    def create_multi_deployment(  # noqa: F811
+        self, config: MultiDeploymentComputeConfig, *, name: Optional[str],
+    ) -> str:
+        full_name, _ = self._private_sdk.create_multi_deployment_compute_config(
+            config, name=name
+        )
+        return full_name
+
     @sdk_docs(
         doc_py_example=_GET_EXAMPLE, arg_docstrings=_GET_ARG_DOCSTRINGS,
     )