anyscale 0.26.0__py3-none-any.whl → 0.26.2__py3-none-any.whl

anyscale/client/openapi_client/models/train_worker.py

@@ -0,0 +1,363 @@
+# 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 TrainWorker(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 = {
+        'actor_id': 'str',
+        'world_rank': 'int',
+        'local_rank': 'int',
+        'node_rank': 'int',
+        'node_id': 'str',
+        'node_ip': 'str',
+        'pid': 'int',
+        'gpu_ids': 'list[int]',
+        'status': 'ActorStatus',
+        'resources': 'TrainResources'
+    }
+
+    attribute_map = {
+        'actor_id': 'actor_id',
+        'world_rank': 'world_rank',
+        'local_rank': 'local_rank',
+        'node_rank': 'node_rank',
+        'node_id': 'node_id',
+        'node_ip': 'node_ip',
+        'pid': 'pid',
+        'gpu_ids': 'gpu_ids',
+        'status': 'status',
+        'resources': 'resources'
+    }
+
+    def __init__(self, actor_id=None, world_rank=None, local_rank=None, node_rank=None, node_id=None, node_ip=None, pid=None, gpu_ids=None, status=None, resources=None, local_vars_configuration=None):  # noqa: E501
+        """TrainWorker - 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._actor_id = None
+        self._world_rank = None
+        self._local_rank = None
+        self._node_rank = None
+        self._node_id = None
+        self._node_ip = None
+        self._pid = None
+        self._gpu_ids = None
+        self._status = None
+        self._resources = None
+        self.discriminator = None
+
+        self.actor_id = actor_id
+        self.world_rank = world_rank
+        self.local_rank = local_rank
+        self.node_rank = node_rank
+        self.node_id = node_id
+        self.node_ip = node_ip
+        self.pid = pid
+        self.gpu_ids = gpu_ids
+        self.status = status
+        if resources is not None:
+            self.resources = resources
+
+    @property
+    def actor_id(self):
+        """Gets the actor_id of this TrainWorker.  # noqa: E501
+
+
+        :return: The actor_id of this TrainWorker.  # noqa: E501
+        :rtype: str
+        """
+        return self._actor_id
+
+    @actor_id.setter
+    def actor_id(self, actor_id):
+        """Sets the actor_id of this TrainWorker.
+
+
+        :param actor_id: The actor_id of this TrainWorker.  # noqa: E501
+        :type: str
+        """
+        if self.local_vars_configuration.client_side_validation and actor_id is None:  # noqa: E501
+            raise ValueError("Invalid value for `actor_id`, must not be `None`")  # noqa: E501
+
+        self._actor_id = actor_id
+
+    @property
+    def world_rank(self):
+        """Gets the world_rank of this TrainWorker.  # noqa: E501
+
+
+        :return: The world_rank of this TrainWorker.  # noqa: E501
+        :rtype: int
+        """
+        return self._world_rank
+
+    @world_rank.setter
+    def world_rank(self, world_rank):
+        """Sets the world_rank of this TrainWorker.
+
+
+        :param world_rank: The world_rank of this TrainWorker.  # noqa: E501
+        :type: int
+        """
+        if self.local_vars_configuration.client_side_validation and world_rank is None:  # noqa: E501
+            raise ValueError("Invalid value for `world_rank`, must not be `None`")  # noqa: E501
+
+        self._world_rank = world_rank
+
+    @property
+    def local_rank(self):
+        """Gets the local_rank of this TrainWorker.  # noqa: E501
+
+
+        :return: The local_rank of this TrainWorker.  # noqa: E501
+        :rtype: int
+        """
+        return self._local_rank
+
+    @local_rank.setter
+    def local_rank(self, local_rank):
+        """Sets the local_rank of this TrainWorker.
+
+
+        :param local_rank: The local_rank of this TrainWorker.  # noqa: E501
+        :type: int
+        """
+        if self.local_vars_configuration.client_side_validation and local_rank is None:  # noqa: E501
+            raise ValueError("Invalid value for `local_rank`, must not be `None`")  # noqa: E501
+
+        self._local_rank = local_rank
+
+    @property
+    def node_rank(self):
+        """Gets the node_rank of this TrainWorker.  # noqa: E501
+
+
+        :return: The node_rank of this TrainWorker.  # noqa: E501
+        :rtype: int
+        """
+        return self._node_rank
+
+    @node_rank.setter
+    def node_rank(self, node_rank):
+        """Sets the node_rank of this TrainWorker.
+
+
+        :param node_rank: The node_rank of this TrainWorker.  # noqa: E501
+        :type: int
+        """
+        if self.local_vars_configuration.client_side_validation and node_rank is None:  # noqa: E501
+            raise ValueError("Invalid value for `node_rank`, must not be `None`")  # noqa: E501
+
+        self._node_rank = node_rank
+
+    @property
+    def node_id(self):
+        """Gets the node_id of this TrainWorker.  # noqa: E501
+
+
+        :return: The node_id of this TrainWorker.  # noqa: E501
+        :rtype: str
+        """
+        return self._node_id
+
+    @node_id.setter
+    def node_id(self, node_id):
+        """Sets the node_id of this TrainWorker.
+
+
+        :param node_id: The node_id of this TrainWorker.  # noqa: E501
+        :type: str
+        """
+        if self.local_vars_configuration.client_side_validation and node_id is None:  # noqa: E501
+            raise ValueError("Invalid value for `node_id`, must not be `None`")  # noqa: E501
+
+        self._node_id = node_id
+
+    @property
+    def node_ip(self):
+        """Gets the node_ip of this TrainWorker.  # noqa: E501
+
+
+        :return: The node_ip of this TrainWorker.  # noqa: E501
+        :rtype: str
+        """
+        return self._node_ip
+
+    @node_ip.setter
+    def node_ip(self, node_ip):
+        """Sets the node_ip of this TrainWorker.
+
+
+        :param node_ip: The node_ip of this TrainWorker.  # noqa: E501
+        :type: str
+        """
+        if self.local_vars_configuration.client_side_validation and node_ip is None:  # noqa: E501
+            raise ValueError("Invalid value for `node_ip`, must not be `None`")  # noqa: E501
+
+        self._node_ip = node_ip
+
+    @property
+    def pid(self):
+        """Gets the pid of this TrainWorker.  # noqa: E501
+
+
+        :return: The pid of this TrainWorker.  # noqa: E501
+        :rtype: int
+        """
+        return self._pid
+
+    @pid.setter
+    def pid(self, pid):
+        """Sets the pid of this TrainWorker.
+
+
+        :param pid: The pid of this TrainWorker.  # noqa: E501
+        :type: int
+        """
+        if self.local_vars_configuration.client_side_validation and pid is None:  # noqa: E501
+            raise ValueError("Invalid value for `pid`, must not be `None`")  # noqa: E501
+
+        self._pid = pid
+
+    @property
+    def gpu_ids(self):
+        """Gets the gpu_ids of this TrainWorker.  # noqa: E501
+
+
+        :return: The gpu_ids of this TrainWorker.  # noqa: E501
+        :rtype: list[int]
+        """
+        return self._gpu_ids
+
+    @gpu_ids.setter
+    def gpu_ids(self, gpu_ids):
+        """Sets the gpu_ids of this TrainWorker.
+
+
+        :param gpu_ids: The gpu_ids of this TrainWorker.  # noqa: E501
+        :type: list[int]
+        """
+        if self.local_vars_configuration.client_side_validation and gpu_ids is None:  # noqa: E501
+            raise ValueError("Invalid value for `gpu_ids`, must not be `None`")  # noqa: E501
+
+        self._gpu_ids = gpu_ids
+
+    @property
+    def status(self):
+        """Gets the status of this TrainWorker.  # noqa: E501
+
+
+        :return: The status of this TrainWorker.  # noqa: E501
+        :rtype: ActorStatus
+        """
+        return self._status
+
+    @status.setter
+    def status(self, status):
+        """Sets the status of this TrainWorker.
+
+
+        :param status: The status of this TrainWorker.  # noqa: E501
+        :type: ActorStatus
+        """
+        if self.local_vars_configuration.client_side_validation and status is None:  # noqa: E501
+            raise ValueError("Invalid value for `status`, must not be `None`")  # noqa: E501
+
+        self._status = status
+
+    @property
+    def resources(self):
+        """Gets the resources of this TrainWorker.  # noqa: E501
+
+
+        :return: The resources of this TrainWorker.  # noqa: E501
+        :rtype: TrainResources
+        """
+        return self._resources
+
+    @resources.setter
+    def resources(self, resources):
+        """Sets the resources of this TrainWorker.
+
+
+        :param resources: The resources of this TrainWorker.  # noqa: E501
+        :type: TrainResources
+        """
+
+        self._resources = resources
+
+    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, TrainWorker):
+            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, TrainWorker):
+            return True
+
+        return self.to_dict() != other.to_dict()

anyscale/client/openapi_client/models/{resourcealert_list_response.py → trainrun_list_response.py}

@@ -18,7 +18,7 @@ import six
 from openapi_client.configuration import Configuration
 
 
-class ResourcealertListResponse(object):
+class TrainrunListResponse(object):
     """NOTE: This class is auto generated by OpenAPI Generator.
     Ref: https://openapi-generator.tech
 
@@ -33,7 +33,7 @@ class ResourcealertListResponse(object):
                             and the value is json key in definition.
     """
     openapi_types = {
-        'results': 'list[ResourceAlert]',
+        'results': 'list[TrainRun]',
         'metadata': 'ListResponseMetadata'
     }
 
@@ -43,7 +43,7 @@ class ResourcealertListResponse(object):
     }
 
     def __init__(self, results=None, metadata=None, local_vars_configuration=None):  # noqa: E501
-        """ResourcealertListResponse - a model defined in OpenAPI"""  # noqa: E501
+        """TrainrunListResponse - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
         self.local_vars_configuration = local_vars_configuration
@@ -58,21 +58,21 @@ class ResourcealertListResponse(object):
 
     @property
     def results(self):
-        """Gets the results of this ResourcealertListResponse.  # noqa: E501
+        """Gets the results of this TrainrunListResponse.  # noqa: E501
 
 
-        :return: The results of this ResourcealertListResponse.  # noqa: E501
-        :rtype: list[ResourceAlert]
+        :return: The results of this TrainrunListResponse.  # noqa: E501
+        :rtype: list[TrainRun]
         """
         return self._results
 
     @results.setter
     def results(self, results):
-        """Sets the results of this ResourcealertListResponse.
+        """Sets the results of this TrainrunListResponse.
 
 
-        :param results: The results of this ResourcealertListResponse.  # noqa: E501
-        :type: list[ResourceAlert]
+        :param results: The results of this TrainrunListResponse.  # noqa: E501
+        :type: list[TrainRun]
         """
         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
@@ -81,20 +81,20 @@ class ResourcealertListResponse(object):
 
     @property
     def metadata(self):
-        """Gets the metadata of this ResourcealertListResponse.  # noqa: E501
+        """Gets the metadata of this TrainrunListResponse.  # noqa: E501
 
 
-        :return: The metadata of this ResourcealertListResponse.  # noqa: E501
+        :return: The metadata of this TrainrunListResponse.  # noqa: E501
         :rtype: ListResponseMetadata
         """
         return self._metadata
 
     @metadata.setter
     def metadata(self, metadata):
-        """Sets the metadata of this ResourcealertListResponse.
+        """Sets the metadata of this TrainrunListResponse.
 
 
-        :param metadata: The metadata of this ResourcealertListResponse.  # noqa: E501
+        :param metadata: The metadata of this TrainrunListResponse.  # noqa: E501
         :type: ListResponseMetadata
         """
 
@@ -134,14 +134,14 @@ class ResourcealertListResponse(object):
 
     def __eq__(self, other):
         """Returns true if both objects are equal"""
-        if not isinstance(other, ResourcealertListResponse):
+        if not isinstance(other, TrainrunListResponse):
             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, ResourcealertListResponse):
+        if not isinstance(other, TrainrunListResponse):
             return True
 
         return self.to_dict() != other.to_dict()

anyscale/client/openapi_client/models/workspace_dataplane_proxied_artifacts.py

@@ -36,17 +36,19 @@ class WorkspaceDataplaneProxiedArtifacts(object):
         'requirements': 'str',
         'skip_packages_tracking': 'str',
         'environment_variables': 'list[str]',
-        'dockerfile': 'str'
+        'dockerfile': 'str',
+        'dockerfile_draft': 'str'
     }
 
     attribute_map = {
         'requirements': 'requirements',
         'skip_packages_tracking': 'skip_packages_tracking',
         'environment_variables': 'environment_variables',
-        'dockerfile': 'dockerfile'
+        'dockerfile': 'dockerfile',
+        'dockerfile_draft': 'dockerfile_draft'
     }
 
-    def __init__(self, requirements=None, skip_packages_tracking=None, environment_variables=None, dockerfile=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, requirements=None, skip_packages_tracking=None, environment_variables=None, dockerfile=None, dockerfile_draft=None, local_vars_configuration=None):  # noqa: E501
         """WorkspaceDataplaneProxiedArtifacts - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -56,6 +58,7 @@ class WorkspaceDataplaneProxiedArtifacts(object):
         self._skip_packages_tracking = None
         self._environment_variables = None
         self._dockerfile = None
+        self._dockerfile_draft = None
         self.discriminator = None
 
         if requirements is not None:
@@ -66,6 +69,8 @@ class WorkspaceDataplaneProxiedArtifacts(object):
             self.environment_variables = environment_variables
         if dockerfile is not None:
             self.dockerfile = dockerfile
+        if dockerfile_draft is not None:
+            self.dockerfile_draft = dockerfile_draft
 
     @property
     def requirements(self):
@@ -159,6 +164,29 @@ class WorkspaceDataplaneProxiedArtifacts(object):
 
         self._dockerfile = dockerfile
 
+    @property
+    def dockerfile_draft(self):
+        """Gets the dockerfile_draft of this WorkspaceDataplaneProxiedArtifacts.  # noqa: E501
+
+        The Dockerfile.draft of the workspace.  # noqa: E501
+
+        :return: The dockerfile_draft of this WorkspaceDataplaneProxiedArtifacts.  # noqa: E501
+        :rtype: str
+        """
+        return self._dockerfile_draft
+
+    @dockerfile_draft.setter
+    def dockerfile_draft(self, dockerfile_draft):
+        """Sets the dockerfile_draft of this WorkspaceDataplaneProxiedArtifacts.
+
+        The Dockerfile.draft of the workspace.  # noqa: E501
+
+        :param dockerfile_draft: The dockerfile_draft of this WorkspaceDataplaneProxiedArtifacts.  # noqa: E501
+        :type: str
+        """
+
+        self._dockerfile_draft = dockerfile_draft
+
     def to_dict(self):
         """Returns the model properties as a dict"""
         result = {}

anyscale/commands/machine_pool_commands.py

@@ -102,6 +102,10 @@ def describe(name: str, format_: str) -> None:
     if format_ == "json":
         print(json.dumps(scheduler_info.to_dict(), default=str))
     elif format_ == "table":
+
+        def format_time(timestamp):
+            return str(timestamp.astimezone().strftime("%m/%d/%Y %I:%M:%S %p %Z"))
+
         machines_table = []
         columns = [
             "MACHINE ID",
@@ -124,11 +128,7 @@ def describe(name: str, format_: str) -> None:
                     if row.workload_info.workload_name
                     else "",
                     row.workload_score,
-                    str(
-                        row.workload_info.workload_start_time.astimezone().strftime(
-                            "%m/%d/%Y %I:%M:%S %p %Z"
-                        )
-                    )
+                    format_time(row.workload_info.workload_start_time)
                     if row.workload_info.workload_name
                     else "",
                     row.cloud_instance_id,
@@ -160,11 +160,7 @@ def describe(name: str, format_: str) -> None:
                     row.size,
                     row.machine_type,
                     f"{row.workload_info.workload_type}/{row.workload_info.workload_name}",
-                    str(
-                        row.workload_info.workload_start_time.astimezone().strftime(
-                            "%m/%d/%Y %I:%M:%S %p %Z"
-                        )
-                    )
+                    format_time(row.workload_info.workload_start_time)
                     if row.workload_info.workload_name
                     else "",
                     row.workload_info.workload_cloud,
@@ -180,6 +176,15 @@ def describe(name: str, format_: str) -> None:
             )
         )
 
+        if (
+            scheduler_info.recent_launch_failures
+            and len(scheduler_info.recent_launch_failures) > 0
+        ):
+            print("Recent Launch Failures:")
+            for failure in scheduler_info.recent_launch_failures:
+                error = failure.error.replace("\n", " ")
+                print(f"- [{format_time(failure.timestamp)}] {error}")
+
 
 @machine_pool_cli.command(
     name="delete",

anyscale/resource_quota/commands.py

@@ -12,7 +12,7 @@ import anyscale
 from anyscale.resource_quota.models import CreateResourceQuota
 
 anyscale.resource_quota.create(
-    ResourceQuota(
+    CreateResourceQuota(
         name="my-resource-quota",
         cloud="my-cloud",
         project="my-project",
@@ -25,7 +25,6 @@ _CREATE_DOCSTRINGS = {"create_resource_quota": "The resource quota to be created
 
 _LIST_EXAMPLE = """
 import anyscale
-from anyscale.resource_quota.models import CreateResourceQuota
 
 anyscale.resource_quota.list(
     name="my-resource-quota",