anyscale 0.26.33__py3-none-any.whl → 0.26.35__py3-none-any.whl

anyscale/_private/workload/workload_sdk.py

@@ -183,7 +183,7 @@ class WorkloadSDK(BaseSDK):
         return new_runtime_envs
 
     def _resolve_compute_config_id(
-        self, compute_config: Union[str, ComputeConfig]
+        self, compute_config: Union[str, ComputeConfig], cloud: Optional[str] = None,
     ) -> str:
         """Resolve the passed compute config to its ID.
 
@@ -195,7 +195,7 @@ class WorkloadSDK(BaseSDK):
         """
         if isinstance(compute_config, str):
             compute_config_id = self._client.get_compute_config_id(
-                compute_config_name=compute_config,
+                compute_config_name=compute_config, cloud=cloud,
             )
             if compute_config_id is None:
                 raise ValueError(
@@ -252,7 +252,8 @@ class WorkloadSDK(BaseSDK):
                 compute_config = replace(compute_config, cloud=cloud)
 
             compute_config_id = self._resolve_compute_config_id(
-                compute_config=compute_config  # type: ignore
+                compute_config=compute_config,  # type: ignore
+                cloud=cloud,
             )
             cloud_id_from_cc = self.client.get_cloud_id(
                 compute_config_id=compute_config_id

anyscale/client/README.md

@@ -331,6 +331,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**sso_login_test_api_v2_users_sso_login_test_get**](docs/DefaultApi.md#sso_login_test_api_v2_users_sso_login_test_get) | **GET** /api/v2/users/sso_login_test | Sso Login Test
 *DefaultApi* | [**start_session_api_v2_sessions_session_id_start_post**](docs/DefaultApi.md#start_session_api_v2_sessions_session_id_start_post) | **POST** /api/v2/sessions/{session_id}/start | Start Session
 *DefaultApi* | [**stop_session_api_v2_sessions_session_id_stop_post**](docs/DefaultApi.md#stop_session_api_v2_sessions_session_id_stop_post) | **POST** /api/v2/sessions/{session_id}/stop | Stop Session
+*DefaultApi* | [**summarize_machine_pool_api_v2_machine_pools_summary_post**](docs/DefaultApi.md#summarize_machine_pool_api_v2_machine_pools_summary_post) | **POST** /api/v2/machine_pools/summary | Summarize Machine Pool
 *DefaultApi* | [**sync_organization_with_metronome_api_v2_organization_billing_organization_id_sync_with_metronome_post**](docs/DefaultApi.md#sync_organization_with_metronome_api_v2_organization_billing_organization_id_sync_with_metronome_post) | **POST** /api/v2/organization_billing/{organization_id}/sync_with_metronome | Sync Organization With Metronome
 *DefaultApi* | [**terminate_job_api_v2_decorated_ha_jobs_production_job_id_terminate_post**](docs/DefaultApi.md#terminate_job_api_v2_decorated_ha_jobs_production_job_id_terminate_post) | **POST** /api/v2/decorated_ha_jobs/{production_job_id}/terminate | Terminate Job
 *DefaultApi* | [**terminate_service_api_v2_services_v2_service_id_terminate_post**](docs/DefaultApi.md#terminate_service_api_v2_services_v2_service_id_terminate_post) | **POST** /api/v2/services-v2/{service_id}/terminate | Terminate Service
@@ -777,8 +778,8 @@ Class | Method | HTTP request | Description
  - [OrganizationinvitationResponse](docs/OrganizationinvitationResponse.md)
  - [OrganizationinvitationbaseResponse](docs/OrganizationinvitationbaseResponse.md)
  - [OrganizationusagealertListResponse](docs/OrganizationusagealertListResponse.md)
- - [PCPConfig](docs/PCPConfig.md)
  - [PageQuery](docs/PageQuery.md)
+ - [PartitionInfo](docs/PartitionInfo.md)
  - [PauseSchedule](docs/PauseSchedule.md)
  - [PermissionLevel](docs/PermissionLevel.md)
  - [ProductionJob](docs/ProductionJob.md)
@@ -878,6 +879,9 @@ Class | Method | HTTP request | Description
  - [StopSessionOptions](docs/StopSessionOptions.md)
  - [StreamPublishRequest](docs/StreamPublishRequest.md)
  - [SubnetIdWithAvailabilityZoneAWS](docs/SubnetIdWithAvailabilityZoneAWS.md)
+ - [SummarizeMachinePoolRequest](docs/SummarizeMachinePoolRequest.md)
+ - [SummarizeMachinePoolResponse](docs/SummarizeMachinePoolResponse.md)
+ - [SummarizemachinepoolresponseResponse](docs/SummarizemachinepoolresponseResponse.md)
  - [SupportRequestsQuery](docs/SupportRequestsQuery.md)
  - [SystemWorkloadName](docs/SystemWorkloadName.md)
  - [TaskExceptionGroupAggregate](docs/TaskExceptionGroupAggregate.md)

anyscale/client/openapi_client/__init__.py

@@ -437,8 +437,8 @@ from openapi_client.models.organizationinvitation_list_response import Organizat
 from openapi_client.models.organizationinvitation_response import OrganizationinvitationResponse
 from openapi_client.models.organizationinvitationbase_response import OrganizationinvitationbaseResponse
 from openapi_client.models.organizationusagealert_list_response import OrganizationusagealertListResponse
-from openapi_client.models.pcp_config import PCPConfig
 from openapi_client.models.page_query import PageQuery
+from openapi_client.models.partition_info import PartitionInfo
 from openapi_client.models.pause_schedule import PauseSchedule
 from openapi_client.models.permission_level import PermissionLevel
 from openapi_client.models.production_job import ProductionJob
@@ -538,6 +538,9 @@ from openapi_client.models.start_session_options import StartSessionOptions
 from openapi_client.models.stop_session_options import StopSessionOptions
 from openapi_client.models.stream_publish_request import StreamPublishRequest
 from openapi_client.models.subnet_id_with_availability_zone_aws import SubnetIdWithAvailabilityZoneAWS
+from openapi_client.models.summarize_machine_pool_request import SummarizeMachinePoolRequest
+from openapi_client.models.summarize_machine_pool_response import SummarizeMachinePoolResponse
+from openapi_client.models.summarizemachinepoolresponse_response import SummarizemachinepoolresponseResponse
 from openapi_client.models.support_requests_query import SupportRequestsQuery
 from openapi_client.models.system_workload_name import SystemWorkloadName
 from openapi_client.models.task_exception_group_aggregate import TaskExceptionGroupAggregate

anyscale/client/openapi_client/api/default_api.py

@@ -9114,8 +9114,8 @@ class DefaultApi(object):
                                                         local_var_params['aggregated_usage_query'] is None):  # noqa: E501
             raise ApiValueError("Missing the required parameter `aggregated_usage_query` when calling `fetch_usage_group_by_cloud_api_v2_aggregated_instance_usage_cloud_post`")  # noqa: E501
 
-        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 50:  # noqa: E501
-            raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_cloud_api_v2_aggregated_instance_usage_cloud_post`, must be a value less than or equal to `50`")  # noqa: E501
+        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 1000:  # noqa: E501
+            raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_cloud_api_v2_aggregated_instance_usage_cloud_post`, must be a value less than or equal to `1000`")  # noqa: E501
         if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] < 0:  # noqa: E501
             raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_cloud_api_v2_aggregated_instance_usage_cloud_post`, must be a value greater than or equal to `0`")  # noqa: E501
         collection_formats = {}
@@ -9244,8 +9244,8 @@ class DefaultApi(object):
                                                         local_var_params['aggregated_usage_query'] is None):  # noqa: E501
             raise ApiValueError("Missing the required parameter `aggregated_usage_query` when calling `fetch_usage_group_by_cluster_api_v2_aggregated_instance_usage_cluster_post`")  # noqa: E501
 
-        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 50:  # noqa: E501
-            raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_cluster_api_v2_aggregated_instance_usage_cluster_post`, must be a value less than or equal to `50`")  # noqa: E501
+        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 1000:  # noqa: E501
+            raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_cluster_api_v2_aggregated_instance_usage_cluster_post`, must be a value less than or equal to `1000`")  # noqa: E501
         if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] < 0:  # noqa: E501
             raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_cluster_api_v2_aggregated_instance_usage_cluster_post`, must be a value greater than or equal to `0`")  # noqa: E501
         collection_formats = {}
@@ -9374,8 +9374,8 @@ class DefaultApi(object):
                                                         local_var_params['aggregated_usage_query'] is None):  # noqa: E501
             raise ApiValueError("Missing the required parameter `aggregated_usage_query` when calling `fetch_usage_group_by_instance_type_api_v2_aggregated_instance_usage_instance_type_post`")  # noqa: E501
 
-        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 50:  # noqa: E501
-            raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_instance_type_api_v2_aggregated_instance_usage_instance_type_post`, must be a value less than or equal to `50`")  # noqa: E501
+        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 1000:  # noqa: E501
+            raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_instance_type_api_v2_aggregated_instance_usage_instance_type_post`, must be a value less than or equal to `1000`")  # noqa: E501
         if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] < 0:  # noqa: E501
             raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_instance_type_api_v2_aggregated_instance_usage_instance_type_post`, must be a value greater than or equal to `0`")  # noqa: E501
         collection_formats = {}
@@ -9504,8 +9504,8 @@ class DefaultApi(object):
                                                         local_var_params['aggregated_usage_query'] is None):  # noqa: E501
             raise ApiValueError("Missing the required parameter `aggregated_usage_query` when calling `fetch_usage_group_by_project_api_v2_aggregated_instance_usage_project_post`")  # noqa: E501
 
-        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 50:  # noqa: E501
-            raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_project_api_v2_aggregated_instance_usage_project_post`, must be a value less than or equal to `50`")  # noqa: E501
+        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 1000:  # noqa: E501
+            raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_project_api_v2_aggregated_instance_usage_project_post`, must be a value less than or equal to `1000`")  # noqa: E501
         if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] < 0:  # noqa: E501
             raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_project_api_v2_aggregated_instance_usage_project_post`, must be a value greater than or equal to `0`")  # noqa: E501
         collection_formats = {}
@@ -9634,8 +9634,8 @@ class DefaultApi(object):
                                                         local_var_params['aggregated_usage_query'] is None):  # noqa: E501
             raise ApiValueError("Missing the required parameter `aggregated_usage_query` when calling `fetch_usage_group_by_user_api_v2_aggregated_instance_usage_user_post`")  # noqa: E501
 
-        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 50:  # noqa: E501
-            raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_user_api_v2_aggregated_instance_usage_user_post`, must be a value less than or equal to `50`")  # noqa: E501
+        if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] > 1000:  # noqa: E501
+            raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_user_api_v2_aggregated_instance_usage_user_post`, must be a value less than or equal to `1000`")  # noqa: E501
         if self.api_client.client_side_validation and 'count' in local_var_params and local_var_params['count'] < 0:  # noqa: E501
             raise ApiValueError("Invalid value for parameter `count` when calling `fetch_usage_group_by_user_api_v2_aggregated_instance_usage_user_post`, must be a value greater than or equal to `0`")  # noqa: E501
         collection_formats = {}
@@ -31055,6 +31055,124 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def summarize_machine_pool_api_v2_machine_pools_summary_post(self, summarize_machine_pool_request, **kwargs):  # noqa: E501
+        """Summarize Machine Pool  # noqa: E501
+
+        Summarize the current state of a machine pool.  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.summarize_machine_pool_api_v2_machine_pools_summary_post(summarize_machine_pool_request, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param SummarizeMachinePoolRequest summarize_machine_pool_request: (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: SummarizemachinepoolresponseResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.summarize_machine_pool_api_v2_machine_pools_summary_post_with_http_info(summarize_machine_pool_request, **kwargs)  # noqa: E501
+
+    def summarize_machine_pool_api_v2_machine_pools_summary_post_with_http_info(self, summarize_machine_pool_request, **kwargs):  # noqa: E501
+        """Summarize Machine Pool  # noqa: E501
+
+        Summarize the current state of a machine pool.  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.summarize_machine_pool_api_v2_machine_pools_summary_post_with_http_info(summarize_machine_pool_request, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param SummarizeMachinePoolRequest summarize_machine_pool_request: (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(SummarizemachinepoolresponseResponse, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'summarize_machine_pool_request'
+        ]
+        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 summarize_machine_pool_api_v2_machine_pools_summary_post" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'summarize_machine_pool_request' is set
+        if self.api_client.client_side_validation and ('summarize_machine_pool_request' not in local_var_params or  # noqa: E501
+                                                        local_var_params['summarize_machine_pool_request'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `summarize_machine_pool_request` when calling `summarize_machine_pool_api_v2_machine_pools_summary_post`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'summarize_machine_pool_request' in local_var_params:
+            body_params = local_var_params['summarize_machine_pool_request']
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # HTTP header `Content-Type`
+        header_params['Content-Type'] = self.api_client.select_header_content_type(  # noqa: E501
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/machine_pools/summary', 'POST',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='SummarizemachinepoolresponseResponse',  # 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 sync_organization_with_metronome_api_v2_organization_billing_organization_id_sync_with_metronome_post(self, organization_id, **kwargs):  # noqa: E501
         """Sync Organization With Metronome  # noqa: E501
 

anyscale/client/openapi_client/models/__init__.py

@@ -423,8 +423,8 @@ from openapi_client.models.organizationinvitation_list_response import Organizat
 from openapi_client.models.organizationinvitation_response import OrganizationinvitationResponse
 from openapi_client.models.organizationinvitationbase_response import OrganizationinvitationbaseResponse
 from openapi_client.models.organizationusagealert_list_response import OrganizationusagealertListResponse
-from openapi_client.models.pcp_config import PCPConfig
 from openapi_client.models.page_query import PageQuery
+from openapi_client.models.partition_info import PartitionInfo
 from openapi_client.models.pause_schedule import PauseSchedule
 from openapi_client.models.permission_level import PermissionLevel
 from openapi_client.models.production_job import ProductionJob
@@ -524,6 +524,9 @@ from openapi_client.models.start_session_options import StartSessionOptions
 from openapi_client.models.stop_session_options import StopSessionOptions
 from openapi_client.models.stream_publish_request import StreamPublishRequest
 from openapi_client.models.subnet_id_with_availability_zone_aws import SubnetIdWithAvailabilityZoneAWS
+from openapi_client.models.summarize_machine_pool_request import SummarizeMachinePoolRequest
+from openapi_client.models.summarize_machine_pool_response import SummarizeMachinePoolResponse
+from openapi_client.models.summarizemachinepoolresponse_response import SummarizemachinepoolresponseResponse
 from openapi_client.models.support_requests_query import SupportRequestsQuery
 from openapi_client.models.system_workload_name import SystemWorkloadName
 from openapi_client.models.task_exception_group_aggregate import TaskExceptionGroupAggregate