anyscale 0.26.60__py3-none-any.whl → 0.26.62__py3-none-any.whl

anyscale/client/openapi_client/models/revokeapikeysresponse_response.py

@@ -0,0 +1,121 @@
+# 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 RevokeapikeysresponseResponse(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 = {
+        'result': 'RevokeApiKeysResponse'
+    }
+
+    attribute_map = {
+        'result': 'result'
+    }
+
+    def __init__(self, result=None, local_vars_configuration=None):  # noqa: E501
+        """RevokeapikeysresponseResponse - 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._result = None
+        self.discriminator = None
+
+        self.result = result
+
+    @property
+    def result(self):
+        """Gets the result of this RevokeapikeysresponseResponse.  # noqa: E501
+
+
+        :return: The result of this RevokeapikeysresponseResponse.  # noqa: E501
+        :rtype: RevokeApiKeysResponse
+        """
+        return self._result
+
+    @result.setter
+    def result(self, result):
+        """Sets the result of this RevokeapikeysresponseResponse.
+
+
+        :param result: The result of this RevokeapikeysresponseResponse.  # noqa: E501
+        :type: RevokeApiKeysResponse
+        """
+        if self.local_vars_configuration.client_side_validation and result is None:  # noqa: E501
+            raise ValueError("Invalid value for `result`, must not be `None`")  # noqa: E501
+
+        self._result = result
+
+    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, RevokeapikeysresponseResponse):
+            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, RevokeapikeysresponseResponse):
+            return True
+
+        return self.to_dict() != other.to_dict()

anyscale/client/openapi_client/models/worker_node_type.py

@@ -36,6 +36,7 @@ class WorkerNodeType(object):
         'name': 'str',
         'instance_type': 'str',
         'resources': 'Resources',
+        'labels': 'dict(str, str)',
         'aws_advanced_configurations_json': 'object',
         'gcp_advanced_configurations_json': 'object',
         'advanced_configurations_json': 'object',
@@ -50,6 +51,7 @@ class WorkerNodeType(object):
         'name': 'name',
         'instance_type': 'instance_type',
         'resources': 'resources',
+        'labels': 'labels',
         'aws_advanced_configurations_json': 'aws_advanced_configurations_json',
         'gcp_advanced_configurations_json': 'gcp_advanced_configurations_json',
         'advanced_configurations_json': 'advanced_configurations_json',
@@ -60,7 +62,7 @@ class WorkerNodeType(object):
         'fallback_to_ondemand': 'fallback_to_ondemand'
     }
 
-    def __init__(self, name=None, instance_type=None, resources=None, aws_advanced_configurations_json=None, gcp_advanced_configurations_json=None, advanced_configurations_json=None, flags=None, min_workers=None, max_workers=None, use_spot=False, fallback_to_ondemand=False, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, name=None, instance_type=None, resources=None, labels=None, aws_advanced_configurations_json=None, gcp_advanced_configurations_json=None, advanced_configurations_json=None, flags=None, min_workers=None, max_workers=None, use_spot=False, fallback_to_ondemand=False, local_vars_configuration=None):  # noqa: E501
         """WorkerNodeType - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -69,6 +71,7 @@ class WorkerNodeType(object):
         self._name = None
         self._instance_type = None
         self._resources = None
+        self._labels = None
         self._aws_advanced_configurations_json = None
         self._gcp_advanced_configurations_json = None
         self._advanced_configurations_json = None
@@ -83,6 +86,8 @@ class WorkerNodeType(object):
         self.instance_type = instance_type
         if resources is not None:
             self.resources = resources
+        if labels is not None:
+            self.labels = labels
         if aws_advanced_configurations_json is not None:
             self.aws_advanced_configurations_json = aws_advanced_configurations_json
         if gcp_advanced_configurations_json is not None:
@@ -173,6 +178,29 @@ class WorkerNodeType(object):
 
         self._resources = resources
 
+    @property
+    def labels(self):
+        """Gets the labels of this WorkerNodeType.  # noqa: E501
+
+        Labels to associate the node with for scheduling purposes. Defaults to the list of Ray & Anyscale default labels.  # noqa: E501
+
+        :return: The labels of this WorkerNodeType.  # noqa: E501
+        :rtype: dict(str, str)
+        """
+        return self._labels
+
+    @labels.setter
+    def labels(self, labels):
+        """Sets the labels of this WorkerNodeType.
+
+        Labels to associate the node with for scheduling purposes. Defaults to the list of Ray & Anyscale default labels.  # noqa: E501
+
+        :param labels: The labels of this WorkerNodeType.  # noqa: E501
+        :type: dict(str, str)
+        """
+
+        self._labels = labels
+
     @property
     def aws_advanced_configurations_json(self):
         """Gets the aws_advanced_configurations_json of this WorkerNodeType.  # noqa: E501

anyscale/client/openapi_client/models/workspace_template.py

@@ -50,7 +50,7 @@ class WorkspaceTemplate(object):
         'updated_at': 'datetime',
         'organization_id': 'str',
         'creator_email': 'str',
-        'published_version': 'WorkspaceTemplateVersionDataObject',
+        'current_version': 'WorkspaceTemplateVersionDataObject',
         'version': 'int'
     }
 
@@ -72,11 +72,11 @@ class WorkspaceTemplate(object):
         'updated_at': 'updated_at',
         'organization_id': 'organization_id',
         'creator_email': 'creator_email',
-        'published_version': 'published_version',
+        'current_version': 'current_version',
         'version': 'version'
     }
 
-    def __init__(self, name=None, description=None, mins_to_complete=None, icon_type=None, icon_bg_color=None, complexity=None, category=None, labels=None, published_version_id=None, cloud_id=None, is_global=False, id=None, creator_id=None, created_at=None, updated_at=None, organization_id=None, creator_email=None, published_version=None, version=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, name=None, description=None, mins_to_complete=None, icon_type=None, icon_bg_color=None, complexity=None, category=None, labels=None, published_version_id=None, cloud_id=None, is_global=False, id=None, creator_id=None, created_at=None, updated_at=None, organization_id=None, creator_email=None, current_version=None, version=None, local_vars_configuration=None):  # noqa: E501
         """WorkspaceTemplate - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -99,7 +99,7 @@ class WorkspaceTemplate(object):
         self._updated_at = None
         self._organization_id = None
         self._creator_email = None
-        self._published_version = None
+        self._current_version = None
         self._version = None
         self.discriminator = None
 
@@ -131,8 +131,8 @@ class WorkspaceTemplate(object):
         self.organization_id = organization_id
         if creator_email is not None:
             self.creator_email = creator_email
-        if published_version is not None:
-            self.published_version = published_version
+        if current_version is not None:
+            self.current_version = current_version
         if version is not None:
             self.version = version
 
@@ -540,27 +540,27 @@ class WorkspaceTemplate(object):
         self._creator_email = creator_email
 
     @property
-    def published_version(self):
-        """Gets the published_version of this WorkspaceTemplate.  # noqa: E501
+    def current_version(self):
+        """Gets the current_version of this WorkspaceTemplate.  # noqa: E501
 
-        Published version of the template  # noqa: E501
+        Current version of the template. This is the version being published or the latest version if no version is published.  # noqa: E501
 
-        :return: The published_version of this WorkspaceTemplate.  # noqa: E501
+        :return: The current_version of this WorkspaceTemplate.  # noqa: E501
         :rtype: WorkspaceTemplateVersionDataObject
         """
-        return self._published_version
+        return self._current_version
 
-    @published_version.setter
-    def published_version(self, published_version):
-        """Sets the published_version of this WorkspaceTemplate.
+    @current_version.setter
+    def current_version(self, current_version):
+        """Sets the current_version of this WorkspaceTemplate.
 
-        Published version of the template  # noqa: E501
+        Current version of the template. This is the version being published or the latest version if no version is published.  # noqa: E501
 
-        :param published_version: The published_version of this WorkspaceTemplate.  # noqa: E501
+        :param current_version: The current_version of this WorkspaceTemplate.  # noqa: E501
         :type: WorkspaceTemplateVersionDataObject
         """
 
-        self._published_version = published_version
+        self._current_version = current_version
 
     @property
     def version(self):

anyscale/compute_config/_private/compute_config_sdk.py

@@ -84,6 +84,7 @@ class PrivateComputeConfigSDK(BaseSDK):
                 resources=self._convert_resource_dict_to_api_model(config.resources)
                 if config.resources is not None or schedulable_by_default
                 else UNSCHEDULABLE_RESOURCES,
+                labels=config.labels,
                 flags=flags or None,
                 advanced_configurations_json=config.advanced_instance_config or None,
             )
@@ -110,6 +111,7 @@ class PrivateComputeConfigSDK(BaseSDK):
                 name=config.name,
                 instance_type=config.instance_type,
                 resources=self._convert_resource_dict_to_api_model(config.resources),
+                labels=config.labels,
                 min_workers=config.min_nodes,
                 max_workers=config.max_nodes,
                 use_spot=config.market_type
@@ -318,6 +320,7 @@ class PrivateComputeConfigSDK(BaseSDK):
         return HeadNodeConfig(
             instance_type=api_model.instance_type,
             resources=self._convert_api_model_to_resource_dict(api_model.resources),
+            labels=api_model.labels,
             advanced_instance_config=self._convert_api_model_to_advanced_instance_config(
                 api_model,
             ),
@@ -365,6 +368,7 @@ class PrivateComputeConfigSDK(BaseSDK):
                     resources=self._convert_api_model_to_resource_dict(
                         api_model.resources
                     ),
+                    labels=api_model.labels,
                     advanced_instance_config=self._convert_api_model_to_advanced_instance_config(
                         api_model,
                     ),

anyscale/compute_config/models.py

@@ -6,6 +6,7 @@ from anyscale._private.models import ModelBase, ModelEnum
 
 
 ResourceDict = Dict[str, float]
+LabelDict = Dict[str, str]
 AdvancedInstanceConfigDict = Dict[str, Any]
 
 
@@ -30,6 +31,18 @@ def _validate_resource_dict(r: Optional[ResourceDict], *, field_name: str):
             )
 
 
+def _validate_label_dict(labels: Optional[LabelDict]):
+    if labels is None:
+        return
+
+    # Convert any non-string keys/values to strings to ensure compatibility
+    for k, v in labels.items():
+        if not isinstance(k, str):
+            raise TypeError(f"'labels' keys must be strings, but got: {k}")
+        if not isinstance(v, str):
+            raise TypeError(f"'labels' values must be strings, but got: {v}")
+
+
 def _validate_advanced_instance_config_dict(c: Optional[AdvancedInstanceConfigDict]):
     if c is None:
         return
@@ -119,6 +132,17 @@ class _NodeConfig(ModelBase):
     def _validate_resources(self, resources: Optional[ResourceDict]):
         _validate_resource_dict(resources, field_name="resources")
 
+    labels: Optional[LabelDict] = field(
+        default=None,
+        repr=False,
+        metadata={
+            "docstring": "Labels to associate the node with for scheduling purposes. Defaults to the list of Ray & Anyscale default labels."
+        },
+    )
+
+    def _validate_labels(self, labels: Optional[LabelDict]):
+        _validate_label_dict(labels)
+
     advanced_instance_config: Optional[AdvancedInstanceConfigDict] = field(
         default=None,
         repr=False,

anyscale/sdk/anyscale_client/models/compute_node_type.py

@@ -36,6 +36,7 @@ class ComputeNodeType(object):
         'name': 'str',
         'instance_type': 'str',
         'resources': 'Resources',
+        'labels': 'dict(str, str)',
         'aws_advanced_configurations_json': 'object',
         'gcp_advanced_configurations_json': 'object',
         'advanced_configurations_json': 'object',
@@ -46,13 +47,14 @@ class ComputeNodeType(object):
         'name': 'name',
         'instance_type': 'instance_type',
         'resources': 'resources',
+        'labels': 'labels',
         'aws_advanced_configurations_json': 'aws_advanced_configurations_json',
         'gcp_advanced_configurations_json': 'gcp_advanced_configurations_json',
         'advanced_configurations_json': 'advanced_configurations_json',
         'flags': 'flags'
     }
 
-    def __init__(self, name=None, instance_type=None, resources=None, aws_advanced_configurations_json=None, gcp_advanced_configurations_json=None, advanced_configurations_json=None, flags=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, name=None, instance_type=None, resources=None, labels=None, aws_advanced_configurations_json=None, gcp_advanced_configurations_json=None, advanced_configurations_json=None, flags=None, local_vars_configuration=None):  # noqa: E501
         """ComputeNodeType - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -61,6 +63,7 @@ class ComputeNodeType(object):
         self._name = None
         self._instance_type = None
         self._resources = None
+        self._labels = None
         self._aws_advanced_configurations_json = None
         self._gcp_advanced_configurations_json = None
         self._advanced_configurations_json = None
@@ -71,6 +74,8 @@ class ComputeNodeType(object):
         self.instance_type = instance_type
         if resources is not None:
             self.resources = resources
+        if labels is not None:
+            self.labels = labels
         if aws_advanced_configurations_json is not None:
             self.aws_advanced_configurations_json = aws_advanced_configurations_json
         if gcp_advanced_configurations_json is not None:
@@ -153,6 +158,29 @@ class ComputeNodeType(object):
 
         self._resources = resources
 
+    @property
+    def labels(self):
+        """Gets the labels of this ComputeNodeType.  # noqa: E501
+
+        Labels to associate the node with for scheduling purposes. Defaults to the list of Ray & Anyscale default labels.  # noqa: E501
+
+        :return: The labels of this ComputeNodeType.  # noqa: E501
+        :rtype: dict(str, str)
+        """
+        return self._labels
+
+    @labels.setter
+    def labels(self, labels):
+        """Sets the labels of this ComputeNodeType.
+
+        Labels to associate the node with for scheduling purposes. Defaults to the list of Ray & Anyscale default labels.  # noqa: E501
+
+        :param labels: The labels of this ComputeNodeType.  # noqa: E501
+        :type: dict(str, str)
+        """
+
+        self._labels = labels
+
     @property
     def aws_advanced_configurations_json(self):
         """Gets the aws_advanced_configurations_json of this ComputeNodeType.  # noqa: E501

anyscale/sdk/anyscale_client/models/worker_node_type.py

@@ -36,6 +36,7 @@ class WorkerNodeType(object):
         'name': 'str',
         'instance_type': 'str',
         'resources': 'Resources',
+        'labels': 'dict(str, str)',
         'aws_advanced_configurations_json': 'object',
         'gcp_advanced_configurations_json': 'object',
         'advanced_configurations_json': 'object',
@@ -50,6 +51,7 @@ class WorkerNodeType(object):
         'name': 'name',
         'instance_type': 'instance_type',
         'resources': 'resources',
+        'labels': 'labels',
         'aws_advanced_configurations_json': 'aws_advanced_configurations_json',
         'gcp_advanced_configurations_json': 'gcp_advanced_configurations_json',
         'advanced_configurations_json': 'advanced_configurations_json',
@@ -60,7 +62,7 @@ class WorkerNodeType(object):
         'fallback_to_ondemand': 'fallback_to_ondemand'
     }
 
-    def __init__(self, name=None, instance_type=None, resources=None, aws_advanced_configurations_json=None, gcp_advanced_configurations_json=None, advanced_configurations_json=None, flags=None, min_workers=None, max_workers=None, use_spot=False, fallback_to_ondemand=False, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, name=None, instance_type=None, resources=None, labels=None, aws_advanced_configurations_json=None, gcp_advanced_configurations_json=None, advanced_configurations_json=None, flags=None, min_workers=None, max_workers=None, use_spot=False, fallback_to_ondemand=False, local_vars_configuration=None):  # noqa: E501
         """WorkerNodeType - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -69,6 +71,7 @@ class WorkerNodeType(object):
         self._name = None
         self._instance_type = None
         self._resources = None
+        self._labels = None
         self._aws_advanced_configurations_json = None
         self._gcp_advanced_configurations_json = None
         self._advanced_configurations_json = None
@@ -83,6 +86,8 @@ class WorkerNodeType(object):
         self.instance_type = instance_type
         if resources is not None:
             self.resources = resources
+        if labels is not None:
+            self.labels = labels
         if aws_advanced_configurations_json is not None:
             self.aws_advanced_configurations_json = aws_advanced_configurations_json
         if gcp_advanced_configurations_json is not None:
@@ -173,6 +178,29 @@ class WorkerNodeType(object):
 
         self._resources = resources
 
+    @property
+    def labels(self):
+        """Gets the labels of this WorkerNodeType.  # noqa: E501
+
+        Labels to associate the node with for scheduling purposes. Defaults to the list of Ray & Anyscale default labels.  # noqa: E501
+
+        :return: The labels of this WorkerNodeType.  # noqa: E501
+        :rtype: dict(str, str)
+        """
+        return self._labels
+
+    @labels.setter
+    def labels(self, labels):
+        """Sets the labels of this WorkerNodeType.
+
+        Labels to associate the node with for scheduling purposes. Defaults to the list of Ray & Anyscale default labels.  # noqa: E501
+
+        :param labels: The labels of this WorkerNodeType.  # noqa: E501
+        :type: dict(str, str)
+        """
+
+        self._labels = labels
+
     @property
     def aws_advanced_configurations_json(self):
         """Gets the aws_advanced_configurations_json of this WorkerNodeType.  # noqa: E501

anyscale/version.py

@@ -1 +1 @@
-__version__ = "0.26.60"
+__version__ = "0.26.62"