anyscale 0.26.42__py3-none-any.whl → 0.26.44__py3-none-any.whl

anyscale/_private/workload/workload_sdk.py

@@ -118,6 +118,7 @@ class WorkloadSDK(BaseSDK):
         autopopulate_in_workspace: bool = True,
         additional_py_modules: Optional[List[str]] = None,
         py_executable_override: Optional[str] = None,
+        has_multiple_cloud_deployments: bool = False,
     ) -> List[Dict[str, Any]]:
         """Returns modified runtime_envs with all local dirs converted to remote URIs.
 
@@ -137,6 +138,11 @@ class WorkloadSDK(BaseSDK):
             if target in local_path_to_uri:
                 return local_path_to_uri[target]
 
+            if has_multiple_cloud_deployments:
+                raise ValueError(
+                    "Local directory uploads are not supported for compute configurations with multiple cloud deployments."
+                )
+
             self.logger.info(f"Uploading local dir '{target}' to cloud storage.")
             assert cloud_id is not None
             uri = self._client.upload_local_dir_to_cloud_storage(

anyscale/client/README.md

@@ -147,6 +147,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**describe_machine_pool_api_v2_machine_pools_describe_post**](docs/DefaultApi.md#describe_machine_pool_api_v2_machine_pools_describe_post) | **POST** /api/v2/machine_pools/describe | Describe Machine Pool
 *DefaultApi* | [**describe_machine_pool_machines_api_v2_machine_pools_describe_machines_post**](docs/DefaultApi.md#describe_machine_pool_machines_api_v2_machine_pools_describe_machines_post) | **POST** /api/v2/machine_pools/describe_machines | Describe Machine Pool Machines
 *DefaultApi* | [**describe_machine_pool_requests_api_v2_machine_pools_describe_requests_post**](docs/DefaultApi.md#describe_machine_pool_requests_api_v2_machine_pools_describe_requests_post) | **POST** /api/v2/machine_pools/describe_requests | Describe Machine Pool Requests
+*DefaultApi* | [**describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post**](docs/DefaultApi.md#describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post) | **POST** /api/v2/machine_pools/describe_workloads | Describe Machine Pool Workloads
 *DefaultApi* | [**describe_system_workload_api_v2_system_workload_cloud_id_describe_post**](docs/DefaultApi.md#describe_system_workload_api_v2_system_workload_cloud_id_describe_post) | **POST** /api/v2/system_workload/{cloud_id}/describe | Describe System Workload
 *DefaultApi* | [**detach_machine_pool_from_cloud_api_v2_machine_pools_detach_post**](docs/DefaultApi.md#detach_machine_pool_from_cloud_api_v2_machine_pools_detach_post) | **POST** /api/v2/machine_pools/detach | Detach Machine Pool From Cloud
 *DefaultApi* | [**download_aggregated_instance_usage_csv_api_v2_aggregated_instance_usage_download_csv_get**](docs/DefaultApi.md#download_aggregated_instance_usage_csv_api_v2_aggregated_instance_usage_download_csv_get) | **GET** /api/v2/aggregated_instance_usage/download_csv | Download Aggregated Instance Usage Csv
@@ -623,10 +624,13 @@ Class | Method | HTTP request | Description
  - [DeletedPlatformFineTunedModel](docs/DeletedPlatformFineTunedModel.md)
  - [DeletedplatformfinetunedmodelResponse](docs/DeletedplatformfinetunedmodelResponse.md)
  - [DeletemachinepoolresponseResponse](docs/DeletemachinepoolresponseResponse.md)
+ - [DescribeMachinePoolMachinesFilters](docs/DescribeMachinePoolMachinesFilters.md)
  - [DescribeMachinePoolMachinesRequest](docs/DescribeMachinePoolMachinesRequest.md)
  - [DescribeMachinePoolRequest](docs/DescribeMachinePoolRequest.md)
  - [DescribeMachinePoolRequestsRequest](docs/DescribeMachinePoolRequestsRequest.md)
  - [DescribeMachinePoolResponse](docs/DescribeMachinePoolResponse.md)
+ - [DescribeMachinePoolWorkloadsFilters](docs/DescribeMachinePoolWorkloadsFilters.md)
+ - [DescribeMachinePoolWorkloadsRequest](docs/DescribeMachinePoolWorkloadsRequest.md)
  - [DescribeSystemWorkloadResponse](docs/DescribeSystemWorkloadResponse.md)
  - [DescribemachinepoolresponseResponse](docs/DescribemachinepoolresponseResponse.md)
  - [DescribesystemworkloadresponseResponse](docs/DescribesystemworkloadresponseResponse.md)
@@ -962,7 +966,10 @@ Class | Method | HTTP request | Description
  - [WandBRunDetails](docs/WandBRunDetails.md)
  - [WorkerNodeType](docs/WorkerNodeType.md)
  - [WorkloadInfo](docs/WorkloadInfo.md)
+ - [WorkloadMachineInfo](docs/WorkloadMachineInfo.md)
+ - [WorkloadStateInfo](docs/WorkloadStateInfo.md)
  - [WorkloadType](docs/WorkloadType.md)
+ - [WorkloadstateinfoListResponse](docs/WorkloadstateinfoListResponse.md)
  - [WorkspaceDataplaneArtifact](docs/WorkspaceDataplaneArtifact.md)
  - [WorkspaceDataplaneArtifacts](docs/WorkspaceDataplaneArtifacts.md)
  - [WorkspaceDataplaneProxiedArtifacts](docs/WorkspaceDataplaneProxiedArtifacts.md)

anyscale/client/openapi_client/__init__.py

@@ -275,10 +275,13 @@ from openapi_client.models.delete_machine_request import DeleteMachineRequest
 from openapi_client.models.deleted_platform_fine_tuned_model import DeletedPlatformFineTunedModel
 from openapi_client.models.deletedplatformfinetunedmodel_response import DeletedplatformfinetunedmodelResponse
 from openapi_client.models.deletemachinepoolresponse_response import DeletemachinepoolresponseResponse
+from openapi_client.models.describe_machine_pool_machines_filters import DescribeMachinePoolMachinesFilters
 from openapi_client.models.describe_machine_pool_machines_request import DescribeMachinePoolMachinesRequest
 from openapi_client.models.describe_machine_pool_request import DescribeMachinePoolRequest
 from openapi_client.models.describe_machine_pool_requests_request import DescribeMachinePoolRequestsRequest
 from openapi_client.models.describe_machine_pool_response import DescribeMachinePoolResponse
+from openapi_client.models.describe_machine_pool_workloads_filters import DescribeMachinePoolWorkloadsFilters
+from openapi_client.models.describe_machine_pool_workloads_request import DescribeMachinePoolWorkloadsRequest
 from openapi_client.models.describe_system_workload_response import DescribeSystemWorkloadResponse
 from openapi_client.models.describemachinepoolresponse_response import DescribemachinepoolresponseResponse
 from openapi_client.models.describesystemworkloadresponse_response import DescribesystemworkloadresponseResponse
@@ -614,7 +617,10 @@ from openapi_client.models.waitliststatusresponse_response import Waitliststatus
 from openapi_client.models.wand_b_run_details import WandBRunDetails
 from openapi_client.models.worker_node_type import WorkerNodeType
 from openapi_client.models.workload_info import WorkloadInfo
+from openapi_client.models.workload_machine_info import WorkloadMachineInfo
+from openapi_client.models.workload_state_info import WorkloadStateInfo
 from openapi_client.models.workload_type import WorkloadType
+from openapi_client.models.workloadstateinfo_list_response import WorkloadstateinfoListResponse
 from openapi_client.models.workspace_dataplane_artifact import WorkspaceDataplaneArtifact
 from openapi_client.models.workspace_dataplane_artifacts import WorkspaceDataplaneArtifacts
 from openapi_client.models.workspace_dataplane_proxied_artifacts import WorkspaceDataplaneProxiedArtifacts

anyscale/client/openapi_client/api/default_api.py

@@ -8419,6 +8419,138 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post(self, describe_machine_pool_workloads_request, **kwargs):  # noqa: E501
+        """Describe Machine Pool Workloads  # noqa: E501
+
+        Describe the currently allocated workloads in 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.describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post(describe_machine_pool_workloads_request, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param DescribeMachinePoolWorkloadsRequest describe_machine_pool_workloads_request: (required)
+        :param str paging_token:
+        :param int count:
+        :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: WorkloadstateinfoListResponse
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post_with_http_info(describe_machine_pool_workloads_request, **kwargs)  # noqa: E501
+
+    def describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post_with_http_info(self, describe_machine_pool_workloads_request, **kwargs):  # noqa: E501
+        """Describe Machine Pool Workloads  # noqa: E501
+
+        Describe the currently allocated workloads in 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.describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post_with_http_info(describe_machine_pool_workloads_request, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param DescribeMachinePoolWorkloadsRequest describe_machine_pool_workloads_request: (required)
+        :param str paging_token:
+        :param int count:
+        :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(WorkloadstateinfoListResponse, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'describe_machine_pool_workloads_request',
+            'paging_token',
+            'count'
+        ]
+        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 describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'describe_machine_pool_workloads_request' is set
+        if self.api_client.client_side_validation and ('describe_machine_pool_workloads_request' not in local_var_params or  # noqa: E501
+                                                        local_var_params['describe_machine_pool_workloads_request'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `describe_machine_pool_workloads_request` when calling `describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post`")  # 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 `describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_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 `describe_machine_pool_workloads_api_v2_machine_pools_describe_workloads_post`, must be a value greater than or equal to `0`")  # noqa: E501
+        collection_formats = {}
+
+        path_params = {}
+
+        query_params = []
+        if 'paging_token' in local_var_params and local_var_params['paging_token'] is not None:  # noqa: E501
+            query_params.append(('paging_token', local_var_params['paging_token']))  # noqa: E501
+        if 'count' in local_var_params and local_var_params['count'] is not None:  # noqa: E501
+            query_params.append(('count', local_var_params['count']))  # noqa: E501
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'describe_machine_pool_workloads_request' in local_var_params:
+            body_params = local_var_params['describe_machine_pool_workloads_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/describe_workloads', 'POST',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='WorkloadstateinfoListResponse',  # 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 describe_system_workload_api_v2_system_workload_cloud_id_describe_post(self, cloud_id, workload_name, **kwargs):  # noqa: E501
         """Describe System Workload  # noqa: E501
 

anyscale/client/openapi_client/models/__init__.py

@@ -261,10 +261,13 @@ from openapi_client.models.delete_machine_request import DeleteMachineRequest
 from openapi_client.models.deleted_platform_fine_tuned_model import DeletedPlatformFineTunedModel
 from openapi_client.models.deletedplatformfinetunedmodel_response import DeletedplatformfinetunedmodelResponse
 from openapi_client.models.deletemachinepoolresponse_response import DeletemachinepoolresponseResponse
+from openapi_client.models.describe_machine_pool_machines_filters import DescribeMachinePoolMachinesFilters
 from openapi_client.models.describe_machine_pool_machines_request import DescribeMachinePoolMachinesRequest
 from openapi_client.models.describe_machine_pool_request import DescribeMachinePoolRequest
 from openapi_client.models.describe_machine_pool_requests_request import DescribeMachinePoolRequestsRequest
 from openapi_client.models.describe_machine_pool_response import DescribeMachinePoolResponse
+from openapi_client.models.describe_machine_pool_workloads_filters import DescribeMachinePoolWorkloadsFilters
+from openapi_client.models.describe_machine_pool_workloads_request import DescribeMachinePoolWorkloadsRequest
 from openapi_client.models.describe_system_workload_response import DescribeSystemWorkloadResponse
 from openapi_client.models.describemachinepoolresponse_response import DescribemachinepoolresponseResponse
 from openapi_client.models.describesystemworkloadresponse_response import DescribesystemworkloadresponseResponse
@@ -600,7 +603,10 @@ from openapi_client.models.waitliststatusresponse_response import Waitliststatus
 from openapi_client.models.wand_b_run_details import WandBRunDetails
 from openapi_client.models.worker_node_type import WorkerNodeType
 from openapi_client.models.workload_info import WorkloadInfo
+from openapi_client.models.workload_machine_info import WorkloadMachineInfo
+from openapi_client.models.workload_state_info import WorkloadStateInfo
 from openapi_client.models.workload_type import WorkloadType
+from openapi_client.models.workloadstateinfo_list_response import WorkloadstateinfoListResponse
 from openapi_client.models.workspace_dataplane_artifact import WorkspaceDataplaneArtifact
 from openapi_client.models.workspace_dataplane_artifacts import WorkspaceDataplaneArtifacts
 from openapi_client.models.workspace_dataplane_proxied_artifacts import WorkspaceDataplaneProxiedArtifacts