anyscale 0.26.35__tar.gz → 0.26.37__tar.gz

{anyscale-0.26.35 → anyscale-0.26.37}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: anyscale
-Version: 0.26.35
+Version: 0.26.37
 Summary: Command Line Interface for Anyscale
 Author: Anyscale Inc.
 License: AS License

{anyscale-0.26.35 → anyscale-0.26.37}/anyscale/_private/docgen/models.md

@@ -990,6 +990,7 @@ Name | Type | Description | Notes
 **cluster_environment_build_id** | **str** | The id of the cluster environment build that will be used to create cluster associated with the queue. | [optional] [default to null]
 **max_concurrency** | **int** | Max number of jobs to be run concurrently. Defaults to 1, ie running no more than 1 job at a time. | [optional] [default to 1]
 **idle_timeout_sec** | **int** | Max period of time queue will be accepting new jobs, before being sealed off and its associated cluster being shutdown | [default to null]
+**auto_termination_threshold_job_count** | **int** | Maximum number of jobs the cluster can run before it becomes eligible for termination. | [optional] [default to null]
 
 ## JobRunType
 
@@ -1398,6 +1399,7 @@ Name | Type | Description | Notes
 **conda** | **object** | [Union[Dict[str, Any], str]: Either the conda YAML config or the name of a local conda env (e.g., \"pytorch_p36\"),  | [optional] [default to null]
 **env_vars** | **Dict(str, str)** | Environment variables to set. | [optional] [default to null]
 **config** | [**object**](#) | Config for runtime environment. Can be used to setup setup_timeout_seconds, the timeout of runtime environment creation. | [optional] [default to null]
+**image_uri** | **str** | Specifies the image URI of the container in which the job will run. | [optional] [default to null]
 
 ## Resources
 

{anyscale-0.26.35 → anyscale-0.26.37}/anyscale/client/README.md

@@ -338,6 +338,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**terminate_system_cluster_api_v2_system_workload_cloud_id_terminate_post**](docs/DefaultApi.md#terminate_system_cluster_api_v2_system_workload_cloud_id_terminate_post) | **POST** /api/v2/system_workload/{cloud_id}/terminate | Terminate System Cluster
 *DefaultApi* | [**test_saml_acs_api_v2_organizations_organization_id_test_saml_acs_post**](docs/DefaultApi.md#test_saml_acs_api_v2_organizations_organization_id_test_saml_acs_post) | **POST** /api/v2/organizations/{organization_id}/test_saml_acs | Test Saml Acs
 *DefaultApi* | [**toggle_instance_usage_budget_is_enabled_api_v2_instance_usage_budgets_instance_usage_budget_id_toggle_is_enabled_post**](docs/DefaultApi.md#toggle_instance_usage_budget_is_enabled_api_v2_instance_usage_budgets_instance_usage_budget_id_toggle_is_enabled_post) | **POST** /api/v2/instance_usage_budgets/{instance_usage_budget_id}/toggle_is_enabled | Toggle Instance Usage Budget Is Enabled
+*DefaultApi* | [**toggle_public_cloud_api_v2_clouds_cloud_id_toggle_public_cloud_put**](docs/DefaultApi.md#toggle_public_cloud_api_v2_clouds_cloud_id_toggle_public_cloud_put) | **PUT** /api/v2/clouds/{cloud_id}/toggle_public_cloud | Toggle Public Cloud
 *DefaultApi* | [**trigger_cron_job_api_v2_experimental_cron_jobs_cron_job_id_trigger_post**](docs/DefaultApi.md#trigger_cron_job_api_v2_experimental_cron_jobs_cron_job_id_trigger_post) | **POST** /api/v2/experimental_cron_jobs/{cron_job_id}/trigger | Trigger Cron Job
 *DefaultApi* | [**try_claim_cloud_api_v2_aioa_cloud_waitlist_claim_cloud_post**](docs/DefaultApi.md#try_claim_cloud_api_v2_aioa_cloud_waitlist_claim_cloud_post) | **POST** /api/v2/aioa_cloud_waitlist/claim_cloud | Try Claim Cloud
 *DefaultApi* | [**try_login_api_v2_users_try_login_post**](docs/DefaultApi.md#try_login_api_v2_users_try_login_post) | **POST** /api/v2/users/try-login | Try Login
@@ -556,6 +557,7 @@ Class | Method | HTTP request | Description
  - [CreditType](docs/CreditType.md)
  - [CreditsV2](docs/CreditsV2.md)
  - [CustomerAlertStatus](docs/CustomerAlertStatus.md)
+ - [DashboardPanel](docs/DashboardPanel.md)
  - [DataplaneServices](docs/DataplaneServices.md)
  - [Dataset](docs/Dataset.md)
  - [DatasetDag](docs/DatasetDag.md)

{anyscale-0.26.35 → anyscale-0.26.37}/anyscale/client/openapi_client/__init__.py

@@ -215,6 +215,7 @@ from openapi_client.models.credit_record_credit_v2 import CreditRecordCreditV2
 from openapi_client.models.credit_type import CreditType
 from openapi_client.models.credits_v2 import CreditsV2
 from openapi_client.models.customer_alert_status import CustomerAlertStatus
+from openapi_client.models.dashboard_panel import DashboardPanel
 from openapi_client.models.dataplane_services import DataplaneServices
 from openapi_client.models.dataset import Dataset
 from openapi_client.models.dataset_dag import DatasetDag

{anyscale-0.26.35 → anyscale-0.26.37}/anyscale/client/openapi_client/api/default_api.py

@@ -31864,6 +31864,118 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def toggle_public_cloud_api_v2_clouds_cloud_id_toggle_public_cloud_put(self, cloud_id, **kwargs):  # noqa: E501
+        """Toggle Public Cloud  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.toggle_public_cloud_api_v2_clouds_cloud_id_toggle_public_cloud_put(cloud_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str cloud_id: (required)
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: object
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.toggle_public_cloud_api_v2_clouds_cloud_id_toggle_public_cloud_put_with_http_info(cloud_id, **kwargs)  # noqa: E501
+
+    def toggle_public_cloud_api_v2_clouds_cloud_id_toggle_public_cloud_put_with_http_info(self, cloud_id, **kwargs):  # noqa: E501
+        """Toggle Public Cloud  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.toggle_public_cloud_api_v2_clouds_cloud_id_toggle_public_cloud_put_with_http_info(cloud_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str cloud_id: (required)
+        :param _return_http_data_only: response data without head status code
+                                       and headers
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: tuple(object, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'cloud_id'
+        ]
+        all_params.extend(
+            [
+                'async_req',
+                '_return_http_data_only',
+                '_preload_content',
+                '_request_timeout'
+            ]
+        )
+
+        for key, val in six.iteritems(local_var_params['kwargs']):
+            if key not in all_params:
+                raise ApiTypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method toggle_public_cloud_api_v2_clouds_cloud_id_toggle_public_cloud_put" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'cloud_id' is set
+        if self.api_client.client_side_validation and ('cloud_id' not in local_var_params or  # noqa: E501
+                                                        local_var_params['cloud_id'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `cloud_id` when calling `toggle_public_cloud_api_v2_clouds_cloud_id_toggle_public_cloud_put`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'cloud_id' in local_var_params:
+            path_params['cloud_id'] = local_var_params['cloud_id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/clouds/{cloud_id}/toggle_public_cloud', 'PUT',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='object',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=local_var_params.get('async_req'),
+            _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
+            _preload_content=local_var_params.get('_preload_content', True),
+            _request_timeout=local_var_params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def trigger_cron_job_api_v2_experimental_cron_jobs_cron_job_id_trigger_post(self, cron_job_id, **kwargs):  # noqa: E501
         """Trigger Cron Job  # noqa: E501
 

{anyscale-0.26.35 → anyscale-0.26.37}/anyscale/client/openapi_client/models/__init__.py

@@ -201,6 +201,7 @@ from openapi_client.models.credit_record_credit_v2 import CreditRecordCreditV2
 from openapi_client.models.credit_type import CreditType
 from openapi_client.models.credits_v2 import CreditsV2
 from openapi_client.models.customer_alert_status import CustomerAlertStatus
+from openapi_client.models.dashboard_panel import DashboardPanel
 from openapi_client.models.dataplane_services import DataplaneServices
 from openapi_client.models.dataset import Dataset
 from openapi_client.models.dataset_dag import DatasetDag

{anyscale-0.26.35 → anyscale-0.26.37}/anyscale/client/openapi_client/models/create_job_queue_requests.py

@@ -39,6 +39,7 @@ class CreateJobQueueRequests(object):
         'cluster_environment_build_id': 'str',
         'max_concurrency': 'int',
         'idle_timeout_sec': 'int',
+        'auto_termination_threshold_job_count': 'int',
         'project_id': 'str'
     }
 
@@ -49,10 +50,11 @@ class CreateJobQueueRequests(object):
         'cluster_environment_build_id': 'cluster_environment_build_id',
         'max_concurrency': 'max_concurrency',
         'idle_timeout_sec': 'idle_timeout_sec',
+        'auto_termination_threshold_job_count': 'auto_termination_threshold_job_count',
         'project_id': 'project_id'
     }
 
-    def __init__(self, job_queue_name=None, execution_mode=None, compute_config_id=None, cluster_environment_build_id=None, max_concurrency=1, idle_timeout_sec=None, project_id=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, job_queue_name=None, execution_mode=None, compute_config_id=None, cluster_environment_build_id=None, max_concurrency=1, idle_timeout_sec=None, auto_termination_threshold_job_count=None, project_id=None, local_vars_configuration=None):  # noqa: E501
         """CreateJobQueueRequests - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -64,6 +66,7 @@ class CreateJobQueueRequests(object):
         self._cluster_environment_build_id = None
         self._max_concurrency = None
         self._idle_timeout_sec = None
+        self._auto_termination_threshold_job_count = None
         self._project_id = None
         self.discriminator = None
 
@@ -76,6 +79,8 @@ class CreateJobQueueRequests(object):
         if max_concurrency is not None:
             self.max_concurrency = max_concurrency
         self.idle_timeout_sec = idle_timeout_sec
+        if auto_termination_threshold_job_count is not None:
+            self.auto_termination_threshold_job_count = auto_termination_threshold_job_count
         self.project_id = project_id
 
     @property
@@ -222,6 +227,29 @@ class CreateJobQueueRequests(object):
 
         self._idle_timeout_sec = idle_timeout_sec
 
+    @property
+    def auto_termination_threshold_job_count(self):
+        """Gets the auto_termination_threshold_job_count of this CreateJobQueueRequests.  # noqa: E501
+
+        Maximum number of jobs the cluster can run before it becomes eligible for termination.  # noqa: E501
+
+        :return: The auto_termination_threshold_job_count of this CreateJobQueueRequests.  # noqa: E501
+        :rtype: int
+        """
+        return self._auto_termination_threshold_job_count
+
+    @auto_termination_threshold_job_count.setter
+    def auto_termination_threshold_job_count(self, auto_termination_threshold_job_count):
+        """Sets the auto_termination_threshold_job_count of this CreateJobQueueRequests.
+
+        Maximum number of jobs the cluster can run before it becomes eligible for termination.  # noqa: E501
+
+        :param auto_termination_threshold_job_count: The auto_termination_threshold_job_count of this CreateJobQueueRequests.  # noqa: E501
+        :type: int
+        """
+
+        self._auto_termination_threshold_job_count = auto_termination_threshold_job_count
+
     @property
     def project_id(self):
         """Gets the project_id of this CreateJobQueueRequests.  # noqa: E501

anyscale-0.26.37/anyscale/client/openapi_client/models/dashboard_panel.py

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

{anyscale-0.26.35 → anyscale-0.26.37}/anyscale/client/openapi_client/models/job_queue_spec.py

@@ -38,7 +38,8 @@ class JobQueueSpec(object):
         'compute_config_id': 'str',
         'cluster_environment_build_id': 'str',
         'max_concurrency': 'int',
-        'idle_timeout_sec': 'int'
+        'idle_timeout_sec': 'int',
+        'auto_termination_threshold_job_count': 'int'
     }
 
     attribute_map = {
@@ -47,10 +48,11 @@ class JobQueueSpec(object):
         'compute_config_id': 'compute_config_id',
         'cluster_environment_build_id': 'cluster_environment_build_id',
         'max_concurrency': 'max_concurrency',
-        'idle_timeout_sec': 'idle_timeout_sec'
+        'idle_timeout_sec': 'idle_timeout_sec',
+        'auto_termination_threshold_job_count': 'auto_termination_threshold_job_count'
     }
 
-    def __init__(self, job_queue_name=None, execution_mode=None, compute_config_id=None, cluster_environment_build_id=None, max_concurrency=1, idle_timeout_sec=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, job_queue_name=None, execution_mode=None, compute_config_id=None, cluster_environment_build_id=None, max_concurrency=1, idle_timeout_sec=None, auto_termination_threshold_job_count=None, local_vars_configuration=None):  # noqa: E501
         """JobQueueSpec - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -62,6 +64,7 @@ class JobQueueSpec(object):
         self._cluster_environment_build_id = None
         self._max_concurrency = None
         self._idle_timeout_sec = None
+        self._auto_termination_threshold_job_count = None
         self.discriminator = None
 
         if job_queue_name is not None:
@@ -75,6 +78,8 @@ class JobQueueSpec(object):
         if max_concurrency is not None:
             self.max_concurrency = max_concurrency
         self.idle_timeout_sec = idle_timeout_sec
+        if auto_termination_threshold_job_count is not None:
+            self.auto_termination_threshold_job_count = auto_termination_threshold_job_count
 
     @property
     def job_queue_name(self):
@@ -216,6 +221,29 @@ class JobQueueSpec(object):
 
         self._idle_timeout_sec = idle_timeout_sec
 
+    @property
+    def auto_termination_threshold_job_count(self):
+        """Gets the auto_termination_threshold_job_count of this JobQueueSpec.  # noqa: E501
+
+        Maximum number of jobs the cluster can run before it becomes eligible for termination.  # noqa: E501
+
+        :return: The auto_termination_threshold_job_count of this JobQueueSpec.  # noqa: E501
+        :rtype: int
+        """
+        return self._auto_termination_threshold_job_count
+
+    @auto_termination_threshold_job_count.setter
+    def auto_termination_threshold_job_count(self, auto_termination_threshold_job_count):
+        """Sets the auto_termination_threshold_job_count of this JobQueueSpec.
+
+        Maximum number of jobs the cluster can run before it becomes eligible for termination.  # noqa: E501
+
+        :param auto_termination_threshold_job_count: The auto_termination_threshold_job_count of this JobQueueSpec.  # noqa: E501
+        :type: int
+        """
+
+        self._auto_termination_threshold_job_count = auto_termination_threshold_job_count
+
     def to_dict(self):
         """Returns the model properties as a dict"""
         result = {}

{anyscale-0.26.35 → anyscale-0.26.37}/anyscale/client/openapi_client/models/ray_runtime_env_config.py

@@ -39,7 +39,8 @@ class RayRuntimeEnvConfig(object):
         'pip': 'list[str]',
         'conda': 'object',
         'env_vars': 'dict(str, str)',
-        'config': 'object'
+        'config': 'object',
+        'image_uri': 'str'
     }
 
     attribute_map = {
@@ -49,10 +50,11 @@ class RayRuntimeEnvConfig(object):
         'pip': 'pip',
         'conda': 'conda',
         'env_vars': 'env_vars',
-        'config': 'config'
+        'config': 'config',
+        'image_uri': 'image_uri'
     }
 
-    def __init__(self, working_dir=None, py_modules=None, py_executable=None, pip=None, conda=None, env_vars=None, config=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, working_dir=None, py_modules=None, py_executable=None, pip=None, conda=None, env_vars=None, config=None, image_uri=None, local_vars_configuration=None):  # noqa: E501
         """RayRuntimeEnvConfig - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -65,6 +67,7 @@ class RayRuntimeEnvConfig(object):
         self._conda = None
         self._env_vars = None
         self._config = None
+        self._image_uri = None
         self.discriminator = None
 
         if working_dir is not None:
@@ -81,6 +84,8 @@ class RayRuntimeEnvConfig(object):
             self.env_vars = env_vars
         if config is not None:
             self.config = config
+        if image_uri is not None:
+            self.image_uri = image_uri
 
     @property
     def working_dir(self):
@@ -243,6 +248,29 @@ class RayRuntimeEnvConfig(object):
 
         self._config = config
 
+    @property
+    def image_uri(self):
+        """Gets the image_uri of this RayRuntimeEnvConfig.  # noqa: E501
+
+        Specifies the image URI of the container in which the job will run.  # noqa: E501
+
+        :return: The image_uri of this RayRuntimeEnvConfig.  # noqa: E501
+        :rtype: str
+        """
+        return self._image_uri
+
+    @image_uri.setter
+    def image_uri(self, image_uri):
+        """Sets the image_uri of this RayRuntimeEnvConfig.
+
+        Specifies the image URI of the container in which the job will run.  # noqa: E501
+
+        :param image_uri: The image_uri of this RayRuntimeEnvConfig.  # noqa: E501
+        :type: str
+        """
+
+        self._image_uri = image_uri
+
     def to_dict(self):
         """Returns the model properties as a dict"""
         result = {}

{anyscale-0.26.35 → anyscale-0.26.37}/anyscale/client/openapi_client/models/train_run.py

@@ -45,7 +45,9 @@ class TrainRun(object):
         'total_attempts': 'int',
         'job_details': 'JobDetails',
         'session_name': 'str',
-        'controller_log_id': 'str'
+        'controller_log_id': 'str',
+        'train_run_panels': 'list[DashboardPanel]',
+        'train_worker_panels': 'list[DashboardPanel]'
     }
 
     attribute_map = {
@@ -61,10 +63,12 @@ class TrainRun(object):
         'total_attempts': 'total_attempts',
         'job_details': 'job_details',
         'session_name': 'session_name',
-        'controller_log_id': 'controller_log_id'
+        'controller_log_id': 'controller_log_id',
+        'train_run_panels': 'train_run_panels',
+        'train_worker_panels': 'train_worker_panels'
     }
 
-    def __init__(self, id=None, name=None, job_id=None, controller_actor_id=None, status=None, status_detail=None, start_time_ms=None, end_time_ms=None, attempts=None, total_attempts=None, job_details=None, session_name=None, controller_log_id=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, id=None, name=None, job_id=None, controller_actor_id=None, status=None, status_detail=None, start_time_ms=None, end_time_ms=None, attempts=None, total_attempts=None, job_details=None, session_name=None, controller_log_id=None, train_run_panels=None, train_worker_panels=None, local_vars_configuration=None):  # noqa: E501
         """TrainRun - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -83,6 +87,8 @@ class TrainRun(object):
         self._job_details = None
         self._session_name = None
         self._controller_log_id = None
+        self._train_run_panels = None
+        self._train_worker_panels = None
         self.discriminator = None
 
         self.id = id
@@ -103,6 +109,10 @@ class TrainRun(object):
             self.session_name = session_name
         if controller_log_id is not None:
             self.controller_log_id = controller_log_id
+        if train_run_panels is not None:
+            self.train_run_panels = train_run_panels
+        if train_worker_panels is not None:
+            self.train_worker_panels = train_worker_panels
 
     @property
     def id(self):
@@ -393,6 +403,48 @@ class TrainRun(object):
 
         self._controller_log_id = controller_log_id
 
+    @property
+    def train_run_panels(self):
+        """Gets the train_run_panels of this TrainRun.  # noqa: E501
+
+
+        :return: The train_run_panels of this TrainRun.  # noqa: E501
+        :rtype: list[DashboardPanel]
+        """
+        return self._train_run_panels
+
+    @train_run_panels.setter
+    def train_run_panels(self, train_run_panels):
+        """Sets the train_run_panels of this TrainRun.
+
+
+        :param train_run_panels: The train_run_panels of this TrainRun.  # noqa: E501
+        :type: list[DashboardPanel]
+        """
+
+        self._train_run_panels = train_run_panels
+
+    @property
+    def train_worker_panels(self):
+        """Gets the train_worker_panels of this TrainRun.  # noqa: E501
+
+
+        :return: The train_worker_panels of this TrainRun.  # noqa: E501
+        :rtype: list[DashboardPanel]
+        """
+        return self._train_worker_panels
+
+    @train_worker_panels.setter
+    def train_worker_panels(self, train_worker_panels):
+        """Sets the train_worker_panels of this TrainRun.
+
+
+        :param train_worker_panels: The train_worker_panels of this TrainRun.  # noqa: E501
+        :type: list[DashboardPanel]
+        """
+
+        self._train_worker_panels = train_worker_panels
+
     def to_dict(self):
         """Returns the model properties as a dict"""
         result = {}

{anyscale-0.26.35 → anyscale-0.26.37}/anyscale/job/_private/job_sdk.py

@@ -190,6 +190,7 @@ class PrivateJobSDK(WorkloadSDK):
                 compute_config_id=compute_config_id,
                 max_concurrency=provided_job_queue_spec.max_concurrency,
                 idle_timeout_sec=provided_job_queue_spec.idle_timeout_s,
+                auto_termination_threshold_job_count=provided_job_queue_spec.auto_termination_threshold_job_count,
             )
 
         job_queue_config = CreateJobQueueConfig(

{anyscale-0.26.35 → anyscale-0.26.37}/anyscale/job/models.py

@@ -133,6 +133,26 @@ job_queue_spec:
                 f"'max_concurrency' must be an int (it is {type(max_concurrency)})."
             )
 
+    auto_termination_threshold_job_count: Optional[int] = field(
+        default=None,
+        metadata={
+            "docstring": "Maximum number of jobs the cluster can run before it "
+            "becomes eligible for termination"
+        },
+    )
+
+    def _validate_auto_termination_threshold_job_count(
+        self, auto_termination_threshold_job_count: Optional[int]
+    ):
+        if auto_termination_threshold_job_count is None:
+            return
+        if not isinstance(auto_termination_threshold_job_count, int):
+            raise TypeError(
+                f"'auto_termination_threshold_job_count' must be an int (it is {type(auto_termination_threshold_job_count)})."
+            )
+        if auto_termination_threshold_job_count <= 0:
+            raise ValueError("'auto_termination_threshold_job_count' should be > 0")
+
 
 @dataclass(frozen=True)
 class JobQueueConfig(ModelBase):