anyscale 0.26.14__py3-none-any.whl → 0.26.16__py3-none-any.whl

anyscale/_private/anyscale_client/anyscale_client.py

@@ -1668,7 +1668,7 @@ class AnyscaleClient(AnyscaleClientInterface):
     def update_workspace(
         self,
         *,
-        workspace_id: str,
+        workspace_id: Optional[str],
         name: Optional[str] = None,
         compute_config_id: Optional[str] = None,
         cluster_environment_build_id: Optional[str] = None,
@@ -1760,7 +1760,9 @@ class AnyscaleClient(AnyscaleClientInterface):
         )
 
     @handle_api_exceptions
-    def get_workspace_cluster(self, workspace_id: str) -> Optional[DecoratedSession]:
+    def get_workspace_cluster(
+        self, workspace_id: Optional[str]
+    ) -> Optional[DecoratedSession]:
         workspace = self.get_workspace(id=workspace_id)
         if not workspace:
             raise ValueError(f"Workspace '{workspace_id}' not found.")

anyscale/_private/anyscale_client/common.py

@@ -561,14 +561,14 @@ class AnyscaleClientInterface(ABC):
 
     @abstractmethod
     def update_workspace_dependencies_offline_only(
-        self, workspace_id: str, requirements: List[str]
+        self, workspace_id: Optional[str], requirements: List[str]
     ):
         """Updates the dynamic dependencies of a workspace while the workspace is offline"""
         raise NotImplementedError
 
     @abstractmethod
     def update_workspace_env_vars_offline_only(
-        self, workspace_id: str, env_vars: Dict[str, str]
+        self, workspace_id: Optional[str], env_vars: Dict[str, str]
     ):
         """Updates the dynamic dependencies of a workspace"""
         raise NotImplementedError
@@ -581,7 +581,9 @@ class AnyscaleClientInterface(ABC):
         raise NotImplementedError
 
     @abstractmethod
-    def get_workspace_cluster(self, workspace_id: str) -> Optional[DecoratedSession]:
+    def get_workspace_cluster(
+        self, workspace_id: Optional[str]
+    ) -> Optional[DecoratedSession]:
         """Get the cluster model for the provided workspace ID."""
         raise NotImplementedError
 
@@ -622,7 +624,7 @@ class AnyscaleClientInterface(ABC):
     def update_workspace(
         self,
         *,
-        workspace_id: str,
+        workspace_id: Optional[str] = None,  # noqa: A002
         name: Optional[str] = None,
         compute_config_id: Optional[str] = None,
         cluster_environment_build_id: Optional[str] = None,

anyscale/_private/anyscale_client/fake_anyscale_client.py

@@ -701,7 +701,10 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         self._job_runs[prod_job_id].append(model)
 
     def register_project_by_name(
-        self, name: str, cloud: str = DEFAULT_CLOUD_NAME, project_id: str = None
+        self,
+        name: str,
+        cloud: str = DEFAULT_CLOUD_NAME,
+        project_id: Optional[str] = None,
     ) -> str:
         """Helper method to create project name to project id mapping."""
         cloud_id = self.get_cloud_id(cloud_name=cloud)
@@ -1063,17 +1066,23 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         workspace.state = SessionState.TERMINATED
 
     def update_workspace_dependencies_offline_only(
-        self, workspace_id: str, requirements: List[str]
+        self, workspace_id: Optional[str], requirements: List[str]
     ):
+        assert workspace_id is not None
         self._workspaces_dependencies[workspace_id] = requirements
 
     def update_workspace_env_vars_offline_only(
-        self, workspace_id: str, env_vars: Dict[str, str]
+        self, workspace_id: Optional[str], env_vars: Dict[str, str]
     ):
+        assert workspace_id is not None
         self._workspaces_env_vars[workspace_id] = env_vars
 
-    def get_workspace_cluster(self, workspace_id: str) -> Optional[DecoratedSession]:
-        workspace_model = self._workspaces.get(workspace_id, None)
+    def get_workspace_cluster(
+        self, workspace_id: Optional[str]
+    ) -> Optional[DecoratedSession]:
+        workspace_model = (
+            self._workspaces.get(workspace_id, None) if workspace_id else None
+        )
         compute_config = self.get_compute_config(workspace_model.compute_config_id)
         assert compute_config is not None
         return Mock(
@@ -1102,7 +1111,7 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         return self._clusters_headnode_ip.get(cluster_id, "")
 
     def get_cluster_ssh_key(self, cluster_id: str) -> SessionSshKey:
-        return self._clusters_ssh_key.get(cluster_id, None)
+        return self._clusters_ssh_key.get(cluster_id, None)  # type: ignore
 
     def get_workspace_default_dir_name(self, workspace_id) -> str:
         workspace = self._workspaces.get(workspace_id, None)
@@ -1136,7 +1145,7 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
     def update_workspace(
         self,
         *,
-        workspace_id: str,
+        workspace_id: Optional[str] = None,  # noqa: A002
         name: Optional[str] = None,
         compute_config_id: Optional[str] = None,
         cluster_environment_build_id: Optional[str] = None,

anyscale/_private/docgen/__main__.py

@@ -233,7 +233,7 @@ ALL_MODULES = [
             "list_schedules": None,
         },
         legacy_cli_prefix="anyscale schedule",
-        legacy_cli_commands={schedule_commands.create, schedule_commands.update,},
+        legacy_cli_commands=[schedule_commands.create, schedule_commands.update],
     ),
     Module(
         title="Service",
@@ -298,7 +298,7 @@ ALL_MODULES = [
             "ServicemodelResponse",
         ],
         legacy_cli_prefix="anyscale service",
-        legacy_cli_commands={service_commands.rollout},
+        legacy_cli_commands=[service_commands.rollout],
     ),
     Module(
         title="Compute Config",
@@ -391,12 +391,12 @@ ALL_MODULES = [
         models=[ImageBuild, ImageBuildStatus],
         legacy_title="Cluster environment",
         legacy_cli_prefix="anyscale image",
-        legacy_cli_commands={
+        legacy_cli_commands=[
             cluster_env_commands.archive,
             cluster_env_commands.build,
             cluster_env_commands.get,
             cluster_env_commands.list,
-        },
+        ],
         legacy_sdk_commands={
             "get_cluster_environment_build_operation": None,
             "create_byod_cluster_environment_build": anyscale.image.build,

anyscale/_private/docgen/generator.py

@@ -60,7 +60,7 @@ class Module:
     legacy_title: Optional[str] = None
     legacy_cli_prefix: Optional[str] = None
     legacy_cli_commands: Optional[List[LegacyAnyscaleCommand]] = None
-    legacy_sdk_commands: Optional[Dict[str, Callable]] = None
+    legacy_sdk_commands: Optional[Dict[str, Optional[Callable]]] = None
     legacy_sdk_models: Optional[List[str]] = None
 
 
@@ -259,7 +259,7 @@ class MarkdownGenerator:
         if typing.get_origin(t) is not None:
             return self._type_container_to_string(t)
         if isinstance(t, (ModelBaseType, ModelEnumType)):
-            return f"[{t.__name__}]({self._get_anchor(t)})"
+            return f"[{t.__name__}]({self._get_anchor(t)})"  # type: ignore
 
         # Avoid poor rendering of unhandled types.
         raise NotImplementedError(

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/sdk/__init__.py

@@ -35,11 +35,11 @@ def sdk_command(
         def wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
             # We disable the mypy linter here because it treats kwargs as a
             # P.kwargs object. mypy wrongly thinks kwargs can't be indexed.
-            if "_sdk" not in kwargs:  # type: ignore
+            if "_private_sdk" not in kwargs:  # type: ignore
                 if key not in _LAZY_SDK_SINGLETONS:
                     _LAZY_SDK_SINGLETONS[key] = sdk_cls()
 
-                kwargs["_sdk"] = _LAZY_SDK_SINGLETONS[key]  # type: ignore
+                kwargs["_private_sdk"] = _LAZY_SDK_SINGLETONS[key]  # type: ignore
 
             return f(*args, **kwargs)
 

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,
             )
@@ -210,7 +211,7 @@ class WorkloadSDK(BaseSDK):
     def resolve_compute_config_and_cloud_id(
         self,
         *,
-        compute_config: Union[None, str, ComputeConfig],
+        compute_config: Union[ComputeConfig, Dict, str, None],
         cloud: Union[None, str],
     ) -> Tuple[str, str]:
         """Resolve the passed compute config to its ID and corresponding cloud ID.
@@ -229,14 +230,14 @@ 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
+                compute_config=compute_config  # type: ignore
             )
             return (
                 compute_config_id,
@@ -251,12 +252,12 @@ class WorkloadSDK(BaseSDK):
                 compute_config = replace(compute_config, cloud=cloud)
 
             compute_config_id = self._resolve_compute_config_id(
-                compute_config=compute_config
+                compute_config=compute_config  # type: ignore
             )
             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."
@@ -269,7 +270,9 @@ class WorkloadSDK(BaseSDK):
         if not self._client.inside_workspace():
             return None
 
-        name = self._client.get_current_workspace_cluster().name
+        workspace = self._client.get_current_workspace_cluster()
+        assert workspace is not None
+        name = workspace.name
         # Defensively default to the workspace cluster name as-is if it doesn't
         # start with the expected prefix.
         if name.startswith(WORKSPACE_CLUSTER_NAME_PREFIX):

anyscale/aggregated_instance_usage/commands.py

@@ -1,3 +1,5 @@
+from typing import Optional
+
 from anyscale._private.sdk import sdk_command
 from anyscale.aggregated_instance_usage._private.aggregated_instance_usage_sdk import (
     PrivateAggregatedInstanceUsageSDK,
@@ -33,8 +35,10 @@ _DOWNLOAD_ARG_DOCSTRINGS = {"filters": "The filter of the instance usage to down
     arg_docstrings=_DOWNLOAD_ARG_DOCSTRINGS,
 )
 def download_csv(
-    filters: DownloadCSVFilters, *, _sdk: PrivateAggregatedInstanceUsageSDK
+    filters: DownloadCSVFilters,
+    *,
+    _private_sdk: Optional[PrivateAggregatedInstanceUsageSDK] = None
 ) -> str:
     """Download an aggregated instance usage report as a zipped CSV to the provided directory.
     """
-    return _sdk.download_csv(filters)
+    return _private_sdk.download_csv(filters)  # type: ignore

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

@@ -172,7 +172,6 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**find_project_by_project_name_api_v2_projects_find_by_name_get**](docs/DefaultApi.md#find_project_by_project_name_api_v2_projects_find_by_name_get) | **GET** /api/v2/projects/find_by_name | Find Project By Project Name
 *DefaultApi* | [**find_with_invitation_api_v2_organizations_find_with_invitation_get**](docs/DefaultApi.md#find_with_invitation_api_v2_organizations_find_with_invitation_get) | **GET** /api/v2/organizations/find_with_invitation | Find With Invitation
 *DefaultApi* | [**find_with_user_api_v2_organizations_find_with_user_get**](docs/DefaultApi.md#find_with_user_api_v2_organizations_find_with_user_get) | **GET** /api/v2/organizations/find_with_user | Find With User
-*DefaultApi* | [**fine_tuning_job_callback_v2_api_v2_fine_tuning_v2_jobs_internal_finish_post**](docs/DefaultApi.md#fine_tuning_job_callback_v2_api_v2_fine_tuning_v2_jobs_internal_finish_post) | **POST** /api/v2/fine_tuning_v2/jobs/internal/finish | Fine Tuning Job Callback V2
 *DefaultApi* | [**finish_session_command_api_v2_session_commands_session_command_id_finish_post**](docs/DefaultApi.md#finish_session_command_api_v2_session_commands_session_command_id_finish_post) | **POST** /api/v2/session_commands/{session_command_id}/finish | Finish Session Command
 *DefaultApi* | [**generate_cloud_data_bucket_presigned_upload_url_api_v2_clouds_cloud_id_generate_cloud_data_bucket_presigned_upload_url_post**](docs/DefaultApi.md#generate_cloud_data_bucket_presigned_upload_url_api_v2_clouds_cloud_id_generate_cloud_data_bucket_presigned_upload_url_post) | **POST** /api/v2/clouds/{cloud_id}/generate_cloud_data_bucket_presigned_upload_url | Generate Cloud Data Bucket Presigned Upload Url
 *DefaultApi* | [**generate_cloud_data_bucket_presigned_url_api_v2_clouds_cloud_id_generate_cloud_data_bucket_presigned_url_post**](docs/DefaultApi.md#generate_cloud_data_bucket_presigned_url_api_v2_clouds_cloud_id_generate_cloud_data_bucket_presigned_url_post) | **POST** /api/v2/clouds/{cloud_id}/generate_cloud_data_bucket_presigned_url | Generate Cloud Data Bucket Presigned Url
@@ -191,7 +190,6 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**get_cloud_with_cloud_resource_api_v2_clouds_with_cloud_resource_router_cloud_id_get**](docs/DefaultApi.md#get_cloud_with_cloud_resource_api_v2_clouds_with_cloud_resource_router_cloud_id_get) | **GET** /api/v2/clouds_with_cloud_resource_router/{cloud_id} | Get Cloud With Cloud Resource
 *DefaultApi* | [**get_cluster_access_token_api_v2_authentication_cluster_id_cluster_access_token_get**](docs/DefaultApi.md#get_cluster_access_token_api_v2_authentication_cluster_id_cluster_access_token_get) | **GET** /api/v2/authentication/{cluster_id}/cluster_access_token | Get Cluster Access Token
 *DefaultApi* | [**get_cluster_events_api_v2_sessions_session_id_cluster_events_get**](docs/DefaultApi.md#get_cluster_events_api_v2_sessions_session_id_cluster_events_get) | **GET** /api/v2/sessions/{session_id}/cluster_events | Get Cluster Events
-*DefaultApi* | [**get_cluster_product_autoscaler_flag_api_v2_logs_cluster_product_autoscaler_flag_session_id_get**](docs/DefaultApi.md#get_cluster_product_autoscaler_flag_api_v2_logs_cluster_product_autoscaler_flag_session_id_get) | **GET** /api/v2/logs/cluster_product_autoscaler_flag/{session_id} | Get Cluster Product Autoscaler Flag
 *DefaultApi* | [**get_compute_template_api_v2_compute_templates_template_id_get**](docs/DefaultApi.md#get_compute_template_api_v2_compute_templates_template_id_get) | **GET** /api/v2/compute_templates/{template_id} | Get Compute Template
 *DefaultApi* | [**get_cron_job_api_v2_experimental_cron_jobs_cron_job_id_get**](docs/DefaultApi.md#get_cron_job_api_v2_experimental_cron_jobs_cron_job_id_get) | **GET** /api/v2/experimental_cron_jobs/{cron_job_id} | Get Cron Job
 *DefaultApi* | [**get_dags_api_v2_dataset_runs_dags_get**](docs/DefaultApi.md#get_dags_api_v2_dataset_runs_dags_get) | **GET** /api/v2/dataset_runs/dags | Get Dags
@@ -199,6 +197,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
@@ -311,6 +310,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**list_organization_project_collaborators_api_v2_projects_project_id_collaborators_organizations_get**](docs/DefaultApi.md#list_organization_project_collaborators_api_v2_projects_project_id_collaborators_organizations_get) | **GET** /api/v2/projects/{project_id}/collaborators/organizations | List Organization Project Collaborators
 *DefaultApi* | [**list_project_collaborators_api_v2_projects_project_id_collaborators_users_get**](docs/DefaultApi.md#list_project_collaborators_api_v2_projects_project_id_collaborators_users_get) | **GET** /api/v2/projects/{project_id}/collaborators/users | List Project Collaborators
 *DefaultApi* | [**list_projects_api_v2_projects_get**](docs/DefaultApi.md#list_projects_api_v2_projects_get) | **GET** /api/v2/projects/ | List Projects
+*DefaultApi* | [**list_ray_sessions_api_v2_tasks_dashboard_list_ray_sessions_get**](docs/DefaultApi.md#list_ray_sessions_api_v2_tasks_dashboard_list_ray_sessions_get) | **GET** /api/v2/tasks_dashboard/list_ray_sessions | List Ray Sessions
 *DefaultApi* | [**list_services_api_v2_services_v2_get**](docs/DefaultApi.md#list_services_api_v2_services_v2_get) | **GET** /api/v2/services-v2/ | List Services
 *DefaultApi* | [**list_sessions_api_v2_sessions_get**](docs/DefaultApi.md#list_sessions_api_v2_sessions_get) | **GET** /api/v2/sessions/ | List Sessions
 *DefaultApi* | [**list_web_terminals_api_v2_sessions_session_id_web_terminals_get**](docs/DefaultApi.md#list_web_terminals_api_v2_sessions_session_id_web_terminals_get) | **GET** /api/v2/sessions/{session_id}/web_terminals | List Web Terminals
@@ -334,7 +334,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
@@ -430,7 +429,6 @@ Class | Method | HTTP request | Description
  - [ArchivedlogsinfoResponse](docs/ArchivedlogsinfoResponse.md)
  - [AttachMachinePoolToCloudRequest](docs/AttachMachinePoolToCloudRequest.md)
  - [AttachmachinepooltocloudresponseResponse](docs/AttachmachinepooltocloudresponseResponse.md)
- - [AviaryModelConfigV2](docs/AviaryModelConfigV2.md)
  - [AwsRegionAndZones](docs/AwsRegionAndZones.md)
  - [AwsRegionInfo](docs/AwsRegionInfo.md)
  - [AwsregionandzonesResponse](docs/AwsregionandzonesResponse.md)
@@ -444,7 +442,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)
@@ -666,7 +663,6 @@ Class | Method | HTTP request | Description
  - [FineTunedModel](docs/FineTunedModel.md)
  - [FinetunedmodelListResponse](docs/FinetunedmodelListResponse.md)
  - [FinetunedmodelResponse](docs/FinetunedmodelResponse.md)
- - [FinishFTJobRequestV2](docs/FinishFTJobRequestV2.md)
  - [GCPFileStoreConfig](docs/GCPFileStoreConfig.md)
  - [GCPMemorystoreInstanceConfig](docs/GCPMemorystoreInstanceConfig.md)
  - [GrafanaDashboard](docs/GrafanaDashboard.md)
@@ -714,11 +710,13 @@ Class | Method | HTTP request | Description
  - [LbresourceResponse](docs/LbresourceResponse.md)
  - [ListMachinePoolsResponse](docs/ListMachinePoolsResponse.md)
  - [ListMachinesResponse](docs/ListMachinesResponse.md)
+ - [ListRaySessionsResponse](docs/ListRaySessionsResponse.md)
  - [ListResourceNotificationsQuery](docs/ListResourceNotificationsQuery.md)
  - [ListResourceQuotasQuery](docs/ListResourceQuotasQuery.md)
  - [ListResponseMetadata](docs/ListResponseMetadata.md)
  - [ListmachinepoolsresponseResponse](docs/ListmachinepoolsresponseResponse.md)
  - [ListmachinesresponseResponse](docs/ListmachinesresponseResponse.md)
+ - [ListraysessionsresponseResponse](docs/ListraysessionsresponseResponse.md)
  - [LogDownloadConfig](docs/LogDownloadConfig.md)
  - [LogDownloadRequest](docs/LogDownloadRequest.md)
  - [LogDownloadResult](docs/LogDownloadResult.md)
@@ -803,9 +801,7 @@ Class | Method | HTTP request | Description
  - [PageQuery](docs/PageQuery.md)
  - [PauseSchedule](docs/PauseSchedule.md)
  - [PermissionLevel](docs/PermissionLevel.md)
- - [ProductAutoscalerFlag](docs/ProductAutoscalerFlag.md)
  - [ProductType](docs/ProductType.md)
- - [ProductautoscalerflagResponse](docs/ProductautoscalerflagResponse.md)
  - [ProductionJob](docs/ProductionJob.md)
  - [ProductionJobConfig](docs/ProductionJobConfig.md)
  - [ProductionJobEvent](docs/ProductionJobEvent.md)
@@ -829,6 +825,7 @@ Class | Method | HTTP request | Description
  - [Quota](docs/Quota.md)
  - [RayGCSExternalStorageConfig](docs/RayGCSExternalStorageConfig.md)
  - [RayRuntimeEnvConfig](docs/RayRuntimeEnvConfig.md)
+ - [RaySession](docs/RaySession.md)
  - [ReadBillingVersion](docs/ReadBillingVersion.md)
  - [ReadbillingversionListResponse](docs/ReadbillingversionListResponse.md)
  - [ReplicaDetails](docs/ReplicaDetails.md)
@@ -913,18 +910,18 @@ Class | Method | HTTP request | Description
  - [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)
- - [TaskexceptiongroupaggregateListResponse](docs/TaskexceptiongroupaggregateListResponse.md)
- - [TaskfunctionnamegroupaggregateListResponse](docs/TaskfunctionnamegroupaggregateListResponse.md)
- - [TaskjobgroupaggregateListResponse](docs/TaskjobgroupaggregateListResponse.md)
  - [TasksummaryResponse](docs/TasksummaryResponse.md)
  - [TemplatizedComputeConfigs](docs/TemplatizedComputeConfigs.md)
  - [TemplatizedDecoratedApplicationTemplates](docs/TemplatizedDecoratedApplicationTemplates.md)

anyscale/client/openapi_client/__init__.py

@@ -61,7 +61,6 @@ from openapi_client.models.archived_logs_info import ArchivedLogsInfo
 from openapi_client.models.archivedlogsinfo_response import ArchivedlogsinfoResponse
 from openapi_client.models.attach_machine_pool_to_cloud_request import AttachMachinePoolToCloudRequest
 from openapi_client.models.attachmachinepooltocloudresponse_response import AttachmachinepooltocloudresponseResponse
-from openapi_client.models.aviary_model_config_v2 import AviaryModelConfigV2
 from openapi_client.models.aws_region_and_zones import AwsRegionAndZones
 from openapi_client.models.aws_region_info import AwsRegionInfo
 from openapi_client.models.awsregionandzones_response import AwsregionandzonesResponse
@@ -75,7 +74,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
@@ -297,7 +295,6 @@ from openapi_client.models.fine_tune_type import FineTuneType
 from openapi_client.models.fine_tuned_model import FineTunedModel
 from openapi_client.models.finetunedmodel_list_response import FinetunedmodelListResponse
 from openapi_client.models.finetunedmodel_response import FinetunedmodelResponse
-from openapi_client.models.finish_ft_job_request_v2 import FinishFTJobRequestV2
 from openapi_client.models.gcp_file_store_config import GCPFileStoreConfig
 from openapi_client.models.gcp_memorystore_instance_config import GCPMemorystoreInstanceConfig
 from openapi_client.models.grafana_dashboard import GrafanaDashboard
@@ -345,11 +342,13 @@ from openapi_client.models.lb_resource import LBResource
 from openapi_client.models.lbresource_response import LbresourceResponse
 from openapi_client.models.list_machine_pools_response import ListMachinePoolsResponse
 from openapi_client.models.list_machines_response import ListMachinesResponse
+from openapi_client.models.list_ray_sessions_response import ListRaySessionsResponse
 from openapi_client.models.list_resource_notifications_query import ListResourceNotificationsQuery
 from openapi_client.models.list_resource_quotas_query import ListResourceQuotasQuery
 from openapi_client.models.list_response_metadata import ListResponseMetadata
 from openapi_client.models.listmachinepoolsresponse_response import ListmachinepoolsresponseResponse
 from openapi_client.models.listmachinesresponse_response import ListmachinesresponseResponse
+from openapi_client.models.listraysessionsresponse_response import ListraysessionsresponseResponse
 from openapi_client.models.log_download_config import LogDownloadConfig
 from openapi_client.models.log_download_request import LogDownloadRequest
 from openapi_client.models.log_download_result import LogDownloadResult
@@ -434,9 +433,7 @@ from openapi_client.models.organizationusagealert_list_response import Organizat
 from openapi_client.models.page_query import PageQuery
 from openapi_client.models.pause_schedule import PauseSchedule
 from openapi_client.models.permission_level import PermissionLevel
-from openapi_client.models.product_autoscaler_flag import ProductAutoscalerFlag
 from openapi_client.models.product_type import ProductType
-from openapi_client.models.productautoscalerflag_response import ProductautoscalerflagResponse
 from openapi_client.models.production_job import ProductionJob
 from openapi_client.models.production_job_config import ProductionJobConfig
 from openapi_client.models.production_job_event import ProductionJobEvent
@@ -460,6 +457,7 @@ from openapi_client.models.python_modules import PythonModules
 from openapi_client.models.quota import Quota
 from openapi_client.models.ray_gcs_external_storage_config import RayGCSExternalStorageConfig
 from openapi_client.models.ray_runtime_env_config import RayRuntimeEnvConfig
+from openapi_client.models.ray_session import RaySession
 from openapi_client.models.read_billing_version import ReadBillingVersion
 from openapi_client.models.readbillingversion_list_response import ReadbillingversionListResponse
 from openapi_client.models.replica_details import ReplicaDetails
@@ -544,18 +542,18 @@ from openapi_client.models.subnet_id_with_availability_zone_aws import SubnetIdW
 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.taskexceptiongroupaggregate_list_response import TaskexceptiongroupaggregateListResponse
-from openapi_client.models.taskfunctionnamegroupaggregate_list_response import TaskfunctionnamegroupaggregateListResponse
-from openapi_client.models.taskjobgroupaggregate_list_response import TaskjobgroupaggregateListResponse
 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