anyscale 0.26.13__py3-none-any.whl → 0.26.15__py3-none-any.whl

anyscale/_private/models/model_base.py

@@ -50,15 +50,18 @@ class ModelEnumType(EnumMeta):
         return cls
 
 
+M = TypeVar("M", bound="ModelEnum")
+
+
 class ModelEnum(str, Enum, metaclass=ModelEnumType):
     # Must be populated by subclasses to document every value.
-    __docstrings__: Dict["ModelEnum", str] = {}
+    __docstrings__: Dict[str, str] = {}
 
     def __str__(self):
         return self.name
 
     @classmethod
-    def validate(cls, value: Union[str, "ModelEnum"]) -> "ModelEnum":
+    def validate(cls: Type[M], value: Union[str, M]) -> M:
         allowed_values = [str(s) for s in cls.__members__ if not str(s).startswith("_")]
         if value.upper() in allowed_values:
             return cls(value.upper())

anyscale/_private/utils/progress_util.py

@@ -1,4 +1,4 @@
-from io import FileIO
+from io import BufferedReader
 import os
 
 from rich.progress import (
@@ -57,7 +57,7 @@ class ProgressFileReader:
     """
 
     def __init__(
-        self, file: FileIO, progress: Progress, task_id: TaskID,
+        self, file: BufferedReader, progress: Progress, task_id: TaskID,
     ):
         self.file = file
         self.progress = progress

anyscale/_private/workload/workload_sdk.py

@@ -138,6 +138,7 @@ class WorkloadSDK(BaseSDK):
                 return local_path_to_uri[target]
 
             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(
                 target, cloud_id=cloud_id, excludes=excludes,
             )
@@ -229,11 +230,11 @@ class WorkloadSDK(BaseSDK):
                 self.client.get_cloud_id(compute_config_id=compute_config_id),
             )
         elif compute_config is None:
-            cloud_id = self.client.get_cloud_id(cloud_name=cloud)
+            cloud_id = self.client.get_cloud_id(cloud_name=cloud)  # type: ignore
             compute_config_id = self._client.get_default_compute_config(
                 cloud_id=cloud_id
             ).id
-            return (compute_config_id, cloud_id)
+            return (compute_config_id, cloud_id)  # type: ignore
         elif cloud is None:
             compute_config_id = self._resolve_compute_config_id(
                 compute_config=compute_config
@@ -256,7 +257,7 @@ class WorkloadSDK(BaseSDK):
             cloud_id_from_cc = self.client.get_cloud_id(
                 compute_config_id=compute_config_id
             )
-            cloud_id_from_cloud = self.client.get_cloud_id(cloud_name=cloud)
+            cloud_id_from_cloud = self.client.get_cloud_id(cloud_name=cloud)  # type: ignore
             if cloud_id_from_cc != cloud_id_from_cloud:
                 raise ValueError(
                     "Cloud does not match from provided `cloud` and `compute_config`. Either pass one of these or ensure they match."

anyscale/anyscale-cloud-setup.yaml

@@ -193,6 +193,10 @@ $SUBNETS_ROUTE_TABLE_ASSOCIATION
               - "s3:DeleteObject"
               - "s3:GetObject"
               - "s3:ListBucket"
+              - "s3:ListBucketMultipartUploads"
+              - "s3:ListMultipartUploadParts"
+              - "s3:AbortMultipartUpload"
+              - "s3:GetBucketLocation"
             Effect: Allow
             Resource:
               - !Sub arn:aws:s3:::${S3Bucket}

anyscale/client/README.md

@@ -199,6 +199,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**get_dataset_api_v2_datasets_dataset_id_get**](docs/DefaultApi.md#get_dataset_api_v2_datasets_dataset_id_get) | **GET** /api/v2/datasets/{dataset_id} | Get Dataset
 *DefaultApi* | [**get_dataset_download_info_api_v2_datasets_download_info_get**](docs/DefaultApi.md#get_dataset_download_info_api_v2_datasets_download_info_get) | **GET** /api/v2/datasets/download_info | Get Dataset Download Info
 *DefaultApi* | [**get_dataset_download_url_api_v2_datasets_download_get**](docs/DefaultApi.md#get_dataset_download_url_api_v2_datasets_download_get) | **GET** /api/v2/datasets/download | Get Dataset Download Url
+*DefaultApi* | [**get_dataset_logs_download_api_v2_dataset_runs_dataset_logs_get**](docs/DefaultApi.md#get_dataset_logs_download_api_v2_dataset_runs_dataset_logs_get) | **GET** /api/v2/dataset_runs/dataset_logs | Get Dataset Logs Download
 *DefaultApi* | [**get_decorated_cluster_api_v2_decorated_sessions_cluster_id_get**](docs/DefaultApi.md#get_decorated_cluster_api_v2_decorated_sessions_cluster_id_get) | **GET** /api/v2/decorated_sessions/{cluster_id} | Get Decorated Cluster
 *DefaultApi* | [**get_decorated_interactive_session_api_v2_decorated_interactive_sessions_interactive_session_id_get**](docs/DefaultApi.md#get_decorated_interactive_session_api_v2_decorated_interactive_sessions_interactive_session_id_get) | **GET** /api/v2/decorated_interactive_sessions/{interactive_session_id} | Get Decorated Interactive Session
 *DefaultApi* | [**get_decorated_job_api_v2_decorated_jobs_job_id_get**](docs/DefaultApi.md#get_decorated_job_api_v2_decorated_jobs_job_id_get) | **GET** /api/v2/decorated_jobs/{job_id} | Get Decorated Job
@@ -260,6 +261,11 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**get_session_ssh_key_api_v2_sessions_session_id_ssh_key_get**](docs/DefaultApi.md#get_session_ssh_key_api_v2_sessions_session_id_ssh_key_get) | **GET** /api/v2/sessions/{session_id}/ssh_key | Get Session Ssh Key
 *DefaultApi* | [**get_startup_logs_api_v2_sessions_session_id_startup_logs_get**](docs/DefaultApi.md#get_startup_logs_api_v2_sessions_session_id_startup_logs_get) | **GET** /api/v2/sessions/{session_id}/startup_logs | Get Startup Logs
 *DefaultApi* | [**get_startup_logs_archived_api_v2_sessions_session_id_startup_logs_archived_get**](docs/DefaultApi.md#get_startup_logs_archived_api_v2_sessions_session_id_startup_logs_archived_get) | **GET** /api/v2/sessions/{session_id}/startup_logs_archived | Get Startup Logs Archived
+*DefaultApi* | [**get_task_exception_aggregates_api_v2_tasks_dashboard_aggregate_by_exception_get**](docs/DefaultApi.md#get_task_exception_aggregates_api_v2_tasks_dashboard_aggregate_by_exception_get) | **GET** /api/v2/tasks_dashboard/aggregate_by_exception | Get Task Exception Aggregates
+*DefaultApi* | [**get_task_function_aggregates_api_v2_tasks_dashboard_aggregate_by_function_get**](docs/DefaultApi.md#get_task_function_aggregates_api_v2_tasks_dashboard_aggregate_by_function_get) | **GET** /api/v2/tasks_dashboard/aggregate_by_function | Get Task Function Aggregates
+*DefaultApi* | [**get_task_job_aggregates_api_v2_tasks_dashboard_aggregate_by_job_get**](docs/DefaultApi.md#get_task_job_aggregates_api_v2_tasks_dashboard_aggregate_by_job_get) | **GET** /api/v2/tasks_dashboard/aggregate_by_job | Get Task Job Aggregates
+*DefaultApi* | [**get_task_summary_api_v2_tasks_dashboard_summary_get**](docs/DefaultApi.md#get_task_summary_api_v2_tasks_dashboard_summary_get) | **GET** /api/v2/tasks_dashboard/summary | Get Task Summary
+*DefaultApi* | [**get_task_table_api_v2_tasks_dashboard_table_get**](docs/DefaultApi.md#get_task_table_api_v2_tasks_dashboard_table_get) | **GET** /api/v2/tasks_dashboard/table | Get Task Table
 *DefaultApi* | [**get_template_readme_api_v2_experimental_workspaces_template_readme_template_id_get**](docs/DefaultApi.md#get_template_readme_api_v2_experimental_workspaces_template_readme_template_id_get) | **GET** /api/v2/experimental_workspaces/template/readme/{template_id} | Get Template Readme
 *DefaultApi* | [**get_templatized_cluster_environments_api_v2_application_templates_templatized_get**](docs/DefaultApi.md#get_templatized_cluster_environments_api_v2_application_templates_templatized_get) | **GET** /api/v2/application_templates/templatized/ | Get Templatized Cluster Environments
 *DefaultApi* | [**get_templatized_compute_configs_api_v2_compute_templates_templatized_cloud_id_get**](docs/DefaultApi.md#get_templatized_compute_configs_api_v2_compute_templates_templatized_cloud_id_get) | **GET** /api/v2/compute_templates/templatized/{cloud_id} | Get Templatized Compute Configs
@@ -329,7 +335,6 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**redirect_to_service_api_v2_sessions_cluster_id_services_get**](docs/DefaultApi.md#redirect_to_service_api_v2_sessions_cluster_id_services_get) | **GET** /api/v2/sessions/{cluster_id}/services | Redirect To Service
 *DefaultApi* | [**redirect_to_tools_api_v2_services_v2_service_id_tools_get**](docs/DefaultApi.md#redirect_to_tools_api_v2_services_v2_service_id_tools_get) | **GET** /api/v2/services-v2/{service_id}/tools | Redirect To Tools
 *DefaultApi* | [**register_api_v2_kubernetes_manager_register_post**](docs/DefaultApi.md#register_api_v2_kubernetes_manager_register_post) | **POST** /api/v2/kubernetes_manager/register | Register
-*DefaultApi* | [**register_buffer_api_v2_anyscaled_register_buffer_post**](docs/DefaultApi.md#register_buffer_api_v2_anyscaled_register_buffer_post) | **POST** /api/v2/anyscaled/register_buffer | Register Buffer
 *DefaultApi* | [**register_build_api_v2_anyscaled_register_build_post**](docs/DefaultApi.md#register_build_api_v2_anyscaled_register_build_post) | **POST** /api/v2/anyscaled/register_build | Register Build
 *DefaultApi* | [**register_machine_api_v2_anyscaled_register_machine_post**](docs/DefaultApi.md#register_machine_api_v2_anyscaled_register_machine_post) | **POST** /api/v2/anyscaled/register_machine | Register Machine
 *DefaultApi* | [**register_user_api_v2_users_post**](docs/DefaultApi.md#register_user_api_v2_users_post) | **POST** /api/v2/users/ | Register User
@@ -439,7 +444,6 @@ Class | Method | HTTP request | Description
  - [BillingInformation](docs/BillingInformation.md)
  - [BillingVersionCode](docs/BillingVersionCode.md)
  - [BodyAwsMarketplaceRegistrationApiV2OrganizationBillingAwsMarketplaceRegistrationPost](docs/BodyAwsMarketplaceRegistrationApiV2OrganizationBillingAwsMarketplaceRegistrationPost.md)
- - [BufferRegistration](docs/BufferRegistration.md)
  - [Build](docs/Build.md)
  - [BuildLogResponse](docs/BuildLogResponse.md)
  - [BuildRegistration](docs/BuildRegistration.md)
@@ -907,6 +911,20 @@ Class | Method | HTTP request | Description
  - [SubnetIdWithAvailabilityZoneAWS](docs/SubnetIdWithAvailabilityZoneAWS.md)
  - [SupportRequestsQuery](docs/SupportRequestsQuery.md)
  - [SystemWorkloadName](docs/SystemWorkloadName.md)
+ - [TaskExceptionGroupAggregate](docs/TaskExceptionGroupAggregate.md)
+ - [TaskExceptionGroupAggregateResponse](docs/TaskExceptionGroupAggregateResponse.md)
+ - [TaskFunctionNameGroupAggregate](docs/TaskFunctionNameGroupAggregate.md)
+ - [TaskFunctionNameGroupAggregateResponse](docs/TaskFunctionNameGroupAggregateResponse.md)
+ - [TaskGroupedAggregateMetrics](docs/TaskGroupedAggregateMetrics.md)
+ - [TaskJobGroupAggregate](docs/TaskJobGroupAggregate.md)
+ - [TaskJobGroupAggregateResponse](docs/TaskJobGroupAggregateResponse.md)
+ - [TaskState](docs/TaskState.md)
+ - [TaskSummary](docs/TaskSummary.md)
+ - [TaskTableConfig](docs/TaskTableConfig.md)
+ - [TaskTableResponse](docs/TaskTableResponse.md)
+ - [TaskTableRow](docs/TaskTableRow.md)
+ - [TaskType](docs/TaskType.md)
+ - [TasksummaryResponse](docs/TasksummaryResponse.md)
  - [TemplatizedComputeConfigs](docs/TemplatizedComputeConfigs.md)
  - [TemplatizedDecoratedApplicationTemplates](docs/TemplatizedDecoratedApplicationTemplates.md)
  - [TemplatizedcomputeconfigsResponse](docs/TemplatizedcomputeconfigsResponse.md)

anyscale/client/openapi_client/__init__.py

@@ -75,7 +75,6 @@ from openapi_client.models.batched_result_organization_invitation_base import Ba
 from openapi_client.models.billing_information import BillingInformation
 from openapi_client.models.billing_version_code import BillingVersionCode
 from openapi_client.models.body_aws_marketplace_registration_api_v2_organization_billing_aws_marketplace_registration_post import BodyAwsMarketplaceRegistrationApiV2OrganizationBillingAwsMarketplaceRegistrationPost
-from openapi_client.models.buffer_registration import BufferRegistration
 from openapi_client.models.build import Build
 from openapi_client.models.build_log_response import BuildLogResponse
 from openapi_client.models.build_registration import BuildRegistration
@@ -543,6 +542,20 @@ 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.support_requests_query import SupportRequestsQuery
 from openapi_client.models.system_workload_name import SystemWorkloadName
+from openapi_client.models.task_exception_group_aggregate import TaskExceptionGroupAggregate
+from openapi_client.models.task_exception_group_aggregate_response import TaskExceptionGroupAggregateResponse
+from openapi_client.models.task_function_name_group_aggregate import TaskFunctionNameGroupAggregate
+from openapi_client.models.task_function_name_group_aggregate_response import TaskFunctionNameGroupAggregateResponse
+from openapi_client.models.task_grouped_aggregate_metrics import TaskGroupedAggregateMetrics
+from openapi_client.models.task_job_group_aggregate import TaskJobGroupAggregate
+from openapi_client.models.task_job_group_aggregate_response import TaskJobGroupAggregateResponse
+from openapi_client.models.task_state import TaskState
+from openapi_client.models.task_summary import TaskSummary
+from openapi_client.models.task_table_config import TaskTableConfig
+from openapi_client.models.task_table_response import TaskTableResponse
+from openapi_client.models.task_table_row import TaskTableRow
+from openapi_client.models.task_type import TaskType
+from openapi_client.models.tasksummary_response import TasksummaryResponse
 from openapi_client.models.templatized_compute_configs import TemplatizedComputeConfigs
 from openapi_client.models.templatized_decorated_application_templates import TemplatizedDecoratedApplicationTemplates
 from openapi_client.models.templatizedcomputeconfigs_response import TemplatizedcomputeconfigsResponse