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

anyscale/client/openapi_client/models/workload_machine_info.py

@@ -0,0 +1,210 @@
+# 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 WorkloadMachineInfo(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 = {
+        'machine_type': 'str',
+        'partition': 'str',
+        'machine_count': 'int',
+        'workload_score': 'int'
+    }
+
+    attribute_map = {
+        'machine_type': 'machine_type',
+        'partition': 'partition',
+        'machine_count': 'machine_count',
+        'workload_score': 'workload_score'
+    }
+
+    def __init__(self, machine_type=None, partition=None, machine_count=None, workload_score=None, local_vars_configuration=None):  # noqa: E501
+        """WorkloadMachineInfo - 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._machine_type = None
+        self._partition = None
+        self._machine_count = None
+        self._workload_score = None
+        self.discriminator = None
+
+        self.machine_type = machine_type
+        self.partition = partition
+        self.machine_count = machine_count
+        self.workload_score = workload_score
+
+    @property
+    def machine_type(self):
+        """Gets the machine_type of this WorkloadMachineInfo.  # noqa: E501
+
+        The type of the machine.  # noqa: E501
+
+        :return: The machine_type of this WorkloadMachineInfo.  # noqa: E501
+        :rtype: str
+        """
+        return self._machine_type
+
+    @machine_type.setter
+    def machine_type(self, machine_type):
+        """Sets the machine_type of this WorkloadMachineInfo.
+
+        The type of the machine.  # noqa: E501
+
+        :param machine_type: The machine_type of this WorkloadMachineInfo.  # noqa: E501
+        :type: str
+        """
+        if self.local_vars_configuration.client_side_validation and machine_type is None:  # noqa: E501
+            raise ValueError("Invalid value for `machine_type`, must not be `None`")  # noqa: E501
+
+        self._machine_type = machine_type
+
+    @property
+    def partition(self):
+        """Gets the partition of this WorkloadMachineInfo.  # noqa: E501
+
+        The partition of the machine.  # noqa: E501
+
+        :return: The partition of this WorkloadMachineInfo.  # noqa: E501
+        :rtype: str
+        """
+        return self._partition
+
+    @partition.setter
+    def partition(self, partition):
+        """Sets the partition of this WorkloadMachineInfo.
+
+        The partition of the machine.  # noqa: E501
+
+        :param partition: The partition of this WorkloadMachineInfo.  # noqa: E501
+        :type: str
+        """
+        if self.local_vars_configuration.client_side_validation and partition is None:  # noqa: E501
+            raise ValueError("Invalid value for `partition`, must not be `None`")  # noqa: E501
+
+        self._partition = partition
+
+    @property
+    def machine_count(self):
+        """Gets the machine_count of this WorkloadMachineInfo.  # noqa: E501
+
+        The number of machines of the given machine type and partition that are allocated to this workload.  # noqa: E501
+
+        :return: The machine_count of this WorkloadMachineInfo.  # noqa: E501
+        :rtype: int
+        """
+        return self._machine_count
+
+    @machine_count.setter
+    def machine_count(self, machine_count):
+        """Sets the machine_count of this WorkloadMachineInfo.
+
+        The number of machines of the given machine type and partition that are allocated to this workload.  # noqa: E501
+
+        :param machine_count: The machine_count of this WorkloadMachineInfo.  # noqa: E501
+        :type: int
+        """
+        if self.local_vars_configuration.client_side_validation and machine_count is None:  # noqa: E501
+            raise ValueError("Invalid value for `machine_count`, must not be `None`")  # noqa: E501
+
+        self._machine_count = machine_count
+
+    @property
+    def workload_score(self):
+        """Gets the workload_score of this WorkloadMachineInfo.  # noqa: E501
+
+        The score of the workload.  # noqa: E501
+
+        :return: The workload_score of this WorkloadMachineInfo.  # noqa: E501
+        :rtype: int
+        """
+        return self._workload_score
+
+    @workload_score.setter
+    def workload_score(self, workload_score):
+        """Sets the workload_score of this WorkloadMachineInfo.
+
+        The score of the workload.  # noqa: E501
+
+        :param workload_score: The workload_score of this WorkloadMachineInfo.  # noqa: E501
+        :type: int
+        """
+        if self.local_vars_configuration.client_side_validation and workload_score is None:  # noqa: E501
+            raise ValueError("Invalid value for `workload_score`, must not be `None`")  # noqa: E501
+
+        self._workload_score = workload_score
+
+    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, WorkloadMachineInfo):
+            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, WorkloadMachineInfo):
+            return True
+
+        return self.to_dict() != other.to_dict()

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."