anyscale 0.26.0__py3-none-any.whl → 0.26.2__py3-none-any.whl

anyscale/client/README.md

@@ -134,7 +134,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**create_organization_collaborator_api_v2_projects_project_id_collaborators_organization_post**](docs/DefaultApi.md#create_organization_collaborator_api_v2_projects_project_id_collaborators_organization_post) | **POST** /api/v2/projects/{project_id}/collaborators/organization | Create Organization Collaborator
 *DefaultApi* | [**create_project_api_v2_projects_post**](docs/DefaultApi.md#create_project_api_v2_projects_post) | **POST** /api/v2/projects/ | Create Project
 *DefaultApi* | [**create_project_collaborator_api_v2_projects_project_id_collaborators_users_post**](docs/DefaultApi.md#create_project_collaborator_api_v2_projects_project_id_collaborators_users_post) | **POST** /api/v2/projects/{project_id}/collaborators/users | Create Project Collaborator
-*DefaultApi* | [**create_resource_alert_api_v2_resource_alerts_post**](docs/DefaultApi.md#create_resource_alert_api_v2_resource_alerts_post) | **POST** /api/v2/resource_alerts/ | Create Resource Alert
+*DefaultApi* | [**create_resource_notification_api_v2_resource_notifications_post**](docs/DefaultApi.md#create_resource_notification_api_v2_resource_notifications_post) | **POST** /api/v2/resource_notifications/ | Create Resource Notification
 *DefaultApi* | [**create_resource_quota_api_v2_resource_quotas_post**](docs/DefaultApi.md#create_resource_quota_api_v2_resource_quotas_post) | **POST** /api/v2/resource_quotas/ | Create Resource Quota
 *DefaultApi* | [**create_service_account_api_v2_users_service_accounts_post**](docs/DefaultApi.md#create_service_account_api_v2_users_service_accounts_post) | **POST** /api/v2/users/service_accounts | Create Service Account
 *DefaultApi* | [**create_session_command_api_v2_sessions_session_id_create_session_command_post**](docs/DefaultApi.md#create_session_command_api_v2_sessions_session_id_create_session_command_post) | **POST** /api/v2/sessions/{session_id}/create_session_command | Create Session Command
@@ -153,7 +153,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**delete_model_api_v2_llm_models_model_id_delete**](docs/DefaultApi.md#delete_model_api_v2_llm_models_model_id_delete) | **DELETE** /api/v2/llm/models/{model_id} | Delete Model
 *DefaultApi* | [**delete_project_api_v2_projects_project_id_delete**](docs/DefaultApi.md#delete_project_api_v2_projects_project_id_delete) | **DELETE** /api/v2/projects/{project_id} | Delete Project
 *DefaultApi* | [**delete_project_collaborator_api_v2_projects_project_id_collaborators_role_or_identity_id_delete**](docs/DefaultApi.md#delete_project_collaborator_api_v2_projects_project_id_collaborators_role_or_identity_id_delete) | **DELETE** /api/v2/projects/{project_id}/collaborators/{role_or_identity_id} | Delete Project Collaborator
-*DefaultApi* | [**delete_resource_alert_api_v2_resource_alerts_resource_alert_id_delete**](docs/DefaultApi.md#delete_resource_alert_api_v2_resource_alerts_resource_alert_id_delete) | **DELETE** /api/v2/resource_alerts/{resource_alert_id} | Delete Resource Alert
+*DefaultApi* | [**delete_resource_notification_api_v2_resource_notifications_resource_notification_id_delete**](docs/DefaultApi.md#delete_resource_notification_api_v2_resource_notifications_resource_notification_id_delete) | **DELETE** /api/v2/resource_notifications/{resource_notification_id} | Delete Resource Notification
 *DefaultApi* | [**delete_resource_quota_api_v2_resource_quotas_resource_quota_id_delete**](docs/DefaultApi.md#delete_resource_quota_api_v2_resource_quotas_resource_quota_id_delete) | **DELETE** /api/v2/resource_quotas/{resource_quota_id} | Delete Resource Quota
 *DefaultApi* | [**delete_session_api_v2_sessions_session_id_delete**](docs/DefaultApi.md#delete_session_api_v2_sessions_session_id_delete) | **DELETE** /api/v2/sessions/{session_id} | Delete Session
 *DefaultApi* | [**delete_web_terminals_api_v2_sessions_session_id_web_terminals_terminal_id_delete**](docs/DefaultApi.md#delete_web_terminals_api_v2_sessions_session_id_web_terminals_terminal_id_delete) | **DELETE** /api/v2/sessions/{session_id}/web_terminals/{terminal_id} | Delete Web Terminals
@@ -264,7 +264,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**get_recent_projects_api_v2_recent_activity_projects_get**](docs/DefaultApi.md#get_recent_projects_api_v2_recent_activity_projects_get) | **GET** /api/v2/recent_activity/projects | Get Recent Projects
 *DefaultApi* | [**get_regions_and_zones_api_v2_aws_info_regions_and_zones_get**](docs/DefaultApi.md#get_regions_and_zones_api_v2_aws_info_regions_and_zones_get) | **GET** /api/v2/aws_info/regions_and_zones | Get Regions And Zones
 *DefaultApi* | [**get_regions_and_zones_api_v2_clouds_gcp_regions_and_zones_get**](docs/DefaultApi.md#get_regions_and_zones_api_v2_clouds_gcp_regions_and_zones_get) | **GET** /api/v2/clouds/gcp/regions_and_zones | Get Regions And Zones
-*DefaultApi* | [**get_resource_alert_api_v2_resource_alerts_resource_alert_id_get**](docs/DefaultApi.md#get_resource_alert_api_v2_resource_alerts_resource_alert_id_get) | **GET** /api/v2/resource_alerts/{resource_alert_id} | Get Resource Alert
+*DefaultApi* | [**get_resource_notification_api_v2_resource_notifications_resource_notification_id_get**](docs/DefaultApi.md#get_resource_notification_api_v2_resource_notifications_resource_notification_id_get) | **GET** /api/v2/resource_notifications/{resource_notification_id} | Get Resource Notification
 *DefaultApi* | [**get_resource_quota_api_v2_resource_quotas_resource_quota_id_get**](docs/DefaultApi.md#get_resource_quota_api_v2_resource_quotas_resource_quota_id_get) | **GET** /api/v2/resource_quotas/{resource_quota_id} | Get Resource Quota
 *DefaultApi* | [**get_runtime_env_logs_api_v2_sessions_session_id_runtime_env_logs_get**](docs/DefaultApi.md#get_runtime_env_logs_api_v2_sessions_session_id_runtime_env_logs_get) | **GET** /api/v2/sessions/{session_id}/runtime_env_logs | Get Runtime Env Logs
 *DefaultApi* | [**get_runtime_env_logs_download_api_v2_logs_runtime_env_logs_download_session_id_get**](docs/DefaultApi.md#get_runtime_env_logs_download_api_v2_logs_runtime_env_logs_download_session_id_get) | **GET** /api/v2/logs/runtime_env_logs_download/{session_id} | Get Runtime Env Logs Download
@@ -291,6 +291,7 @@ Class | Method | HTTP request | Description
 *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
 *DefaultApi* | [**get_total_usage_api_v2_aggregated_instance_usage_get**](docs/DefaultApi.md#get_total_usage_api_v2_aggregated_instance_usage_get) | **GET** /api/v2/aggregated_instance_usage/ | Get Total Usage
+*DefaultApi* | [**get_train_runs_api_v2_train_runs_get**](docs/DefaultApi.md#get_train_runs_api_v2_train_runs_get) | **GET** /api/v2/train_runs/ | Get Train Runs
 *DefaultApi* | [**get_user_info_api_v2_userinfo_get**](docs/DefaultApi.md#get_user_info_api_v2_userinfo_get) | **GET** /api/v2/userinfo/ | Get User Info
 *DefaultApi* | [**get_workspace_api_v2_experimental_workspaces_workspace_id_get**](docs/DefaultApi.md#get_workspace_api_v2_experimental_workspaces_workspace_id_get) | **GET** /api/v2/experimental_workspaces/{workspace_id} | Get Workspace
 *DefaultApi* | [**get_workspace_dataplane_artifacts_api_v2_experimental_workspaces_workspace_id_dataplane_artifacts_get**](docs/DefaultApi.md#get_workspace_dataplane_artifacts_api_v2_experimental_workspaces_workspace_id_dataplane_artifacts_get) | **GET** /api/v2/experimental_workspaces/{workspace_id}/dataplane_artifacts | Get Workspace Dataplane Artifacts
@@ -383,7 +384,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**search_cluster_environments_api_v2_application_templates_search_post**](docs/DefaultApi.md#search_cluster_environments_api_v2_application_templates_search_post) | **POST** /api/v2/application_templates/search | Search Cluster Environments
 *DefaultApi* | [**search_compute_templates_api_v2_compute_templates_search_post**](docs/DefaultApi.md#search_compute_templates_api_v2_compute_templates_search_post) | **POST** /api/v2/compute_templates/search | Search Compute Templates
 *DefaultApi* | [**search_invoices_api_v2_organization_billing_invoices_post**](docs/DefaultApi.md#search_invoices_api_v2_organization_billing_invoices_post) | **POST** /api/v2/organization_billing/invoices | Search Invoices
-*DefaultApi* | [**search_resource_alerts_api_v2_resource_alerts_search_post**](docs/DefaultApi.md#search_resource_alerts_api_v2_resource_alerts_search_post) | **POST** /api/v2/resource_alerts/search | Search Resource Alerts
+*DefaultApi* | [**search_resource_notifications_api_v2_resource_notifications_search_post**](docs/DefaultApi.md#search_resource_notifications_api_v2_resource_notifications_search_post) | **POST** /api/v2/resource_notifications/search | Search Resource Notifications
 *DefaultApi* | [**search_resource_quotas_api_v2_resource_quotas_search_post**](docs/DefaultApi.md#search_resource_quotas_api_v2_resource_quotas_search_post) | **POST** /api/v2/resource_quotas/search | Search Resource Quotas
 *DefaultApi* | [**search_support_requests_api_v2_support_requests_search_post**](docs/DefaultApi.md#search_support_requests_api_v2_support_requests_search_post) | **POST** /api/v2/support_requests/search | Search Support Requests
 *DefaultApi* | [**search_user_cards_api_v2_onboarding_cards_search_post**](docs/DefaultApi.md#search_user_cards_api_v2_onboarding_cards_search_post) | **POST** /api/v2/onboarding_cards/search | Search User Cards
@@ -417,6 +418,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**update_instance_usage_budget_api_v2_instance_usage_budgets_instance_usage_budget_id_put**](docs/DefaultApi.md#update_instance_usage_budget_api_v2_instance_usage_budgets_instance_usage_budget_id_put) | **PUT** /api/v2/instance_usage_budgets/{instance_usage_budget_id} | Update Instance Usage Budget
 *DefaultApi* | [**update_job_queue_config_api_v2_decorated_ha_jobs_production_job_id_update_job_queue_config_put**](docs/DefaultApi.md#update_job_queue_config_api_v2_decorated_ha_jobs_production_job_id_update_job_queue_config_put) | **PUT** /api/v2/decorated_ha_jobs/{production_job_id}/update_job_queue_config | Update Job Queue Config
 *DefaultApi* | [**update_machine_pool_api_v2_machine_pools_update_post**](docs/DefaultApi.md#update_machine_pool_api_v2_machine_pools_update_post) | **POST** /api/v2/machine_pools/update | Update Machine Pool
+*DefaultApi* | [**update_resource_notification_api_v2_resource_notifications_resource_notification_id_put**](docs/DefaultApi.md#update_resource_notification_api_v2_resource_notifications_resource_notification_id_put) | **PUT** /api/v2/resource_notifications/{resource_notification_id} | Update Resource Notification
 *DefaultApi* | [**update_resource_quota_api_v2_resource_quotas_resource_quota_id_patch**](docs/DefaultApi.md#update_resource_quota_api_v2_resource_quotas_resource_quota_id_patch) | **PATCH** /api/v2/resource_quotas/{resource_quota_id} | Update Resource Quota
 *DefaultApi* | [**upload_session_command_logs_api_v2_session_commands_session_command_id_upload_logs_post**](docs/DefaultApi.md#upload_session_command_logs_api_v2_session_commands_session_command_id_upload_logs_post) | **POST** /api/v2/session_commands/{session_command_id}/upload_logs | Upload Session Command Logs
 *DefaultApi* | [**upsert_billing_version_api_v2_organization_billing_billing_versions_put**](docs/DefaultApi.md#upsert_billing_version_api_v2_organization_billing_billing_versions_put) | **PUT** /api/v2/organization_billing/billing_versions | Upsert Billing Version
@@ -436,6 +438,7 @@ Class | Method | HTTP request | Description
  - [AWSCredentials](docs/AWSCredentials.md)
  - [AWSMemoryDBClusterConfig](docs/AWSMemoryDBClusterConfig.md)
  - [AccessConfig](docs/AccessConfig.md)
+ - [ActorStatus](docs/ActorStatus.md)
  - [AdminCreateUser](docs/AdminCreateUser.md)
  - [AdminCreatedUser](docs/AdminCreatedUser.md)
  - [AdmincreateduserListResponse](docs/AdmincreateduserListResponse.md)
@@ -621,7 +624,7 @@ Class | Method | HTTP request | Description
  - [CreateOrganizationConfiguration](docs/CreateOrganizationConfiguration.md)
  - [CreateOrganizationInvitation](docs/CreateOrganizationInvitation.md)
  - [CreateProductionJobConfig](docs/CreateProductionJobConfig.md)
- - [CreateResourceAlert](docs/CreateResourceAlert.md)
+ - [CreateResourceNotification](docs/CreateResourceNotification.md)
  - [CreateResourceQuota](docs/CreateResourceQuota.md)
  - [CreateSchedule](docs/CreateSchedule.md)
  - [CreateSessionFromSnapshotOptions](docs/CreateSessionFromSnapshotOptions.md)
@@ -639,7 +642,6 @@ Class | Method | HTTP request | Description
  - [CreatesessionresponseResponse](docs/CreatesessionresponseResponse.md)
  - [CreditCardInformation](docs/CreditCardInformation.md)
  - [CustomerAlertStatus](docs/CustomerAlertStatus.md)
- - [CustomerBillingType](docs/CustomerBillingType.md)
  - [DataplaneServices](docs/DataplaneServices.md)
  - [Dataset](docs/Dataset.md)
  - [DatasetListResponse](docs/DatasetListResponse.md)
@@ -765,6 +767,7 @@ Class | Method | HTTP request | Description
  - [InvoiceStatus](docs/InvoiceStatus.md)
  - [InvoicesQuery](docs/InvoicesQuery.md)
  - [JobAccess](docs/JobAccess.md)
+ - [JobDetails](docs/JobDetails.md)
  - [JobQueue](docs/JobQueue.md)
  - [JobQueueConfig](docs/JobQueueConfig.md)
  - [JobQueueExecutionMode](docs/JobQueueExecutionMode.md)
@@ -789,7 +792,7 @@ Class | Method | HTTP request | Description
  - [LbresourceResponse](docs/LbresourceResponse.md)
  - [ListMachinePoolsResponse](docs/ListMachinePoolsResponse.md)
  - [ListMachinesResponse](docs/ListMachinesResponse.md)
- - [ListResourceAlertsQuery](docs/ListResourceAlertsQuery.md)
+ - [ListResourceNotificationsQuery](docs/ListResourceNotificationsQuery.md)
  - [ListResourceQuotasQuery](docs/ListResourceQuotasQuery.md)
  - [ListResponseMetadata](docs/ListResponseMetadata.md)
  - [ListmachinepoolsresponseResponse](docs/ListmachinepoolsresponseResponse.md)
@@ -816,6 +819,7 @@ Class | Method | HTTP request | Description
  - [MachineAllocationState](docs/MachineAllocationState.md)
  - [MachineConnectionState](docs/MachineConnectionState.md)
  - [MachineInfo](docs/MachineInfo.md)
+ - [MachineLaunchFailure](docs/MachineLaunchFailure.md)
  - [MachinePool](docs/MachinePool.md)
  - [MachineStateInfo](docs/MachineStateInfo.md)
  - [MetronomeCustomerInfoModel](docs/MetronomeCustomerInfoModel.md)
@@ -925,17 +929,19 @@ Class | Method | HTTP request | Description
  - [RequestemailmagiclinkresponseResponse](docs/RequestemailmagiclinkresponseResponse.md)
  - [RequestotpreturnapimodelResponse](docs/RequestotpreturnapimodelResponse.md)
  - [ResetPasswordParams](docs/ResetPasswordParams.md)
- - [ResourceAlert](docs/ResourceAlert.md)
  - [ResourceAlertEventType](docs/ResourceAlertEventType.md)
+ - [ResourceNotification](docs/ResourceNotification.md)
  - [ResourceQuota](docs/ResourceQuota.md)
  - [ResourceQuotaStatus](docs/ResourceQuotaStatus.md)
- - [ResourcealertListResponse](docs/ResourcealertListResponse.md)
- - [ResourcealertResponse](docs/ResourcealertResponse.md)
+ - [ResourcenotificationListResponse](docs/ResourcenotificationListResponse.md)
+ - [ResourcenotificationResponse](docs/ResourcenotificationResponse.md)
  - [ResourcequotaListResponse](docs/ResourcequotaListResponse.md)
  - [ResourcequotaResponse](docs/ResourcequotaResponse.md)
  - [Resources](docs/Resources.md)
  - [RollbackServiceModel](docs/RollbackServiceModel.md)
  - [RolloutStrategy](docs/RolloutStrategy.md)
+ - [RunAttemptStatus](docs/RunAttemptStatus.md)
+ - [RunStatus](docs/RunStatus.md)
  - [S3DownloadLocation](docs/S3DownloadLocation.md)
  - [SSOLoginInfo](docs/SSOLoginInfo.md)
  - [SUPPORTEDBASEIMAGESENUM](docs/SUPPORTEDBASEIMAGESENUM.md)
@@ -1021,6 +1027,11 @@ Class | Method | HTTP request | Description
  - [TimestampedlogsoutputResponse](docs/TimestampedlogsoutputResponse.md)
  - [Tool](docs/Tool.md)
  - [TracingConfig](docs/TracingConfig.md)
+ - [TrainResources](docs/TrainResources.md)
+ - [TrainRun](docs/TrainRun.md)
+ - [TrainRunAttempt](docs/TrainRunAttempt.md)
+ - [TrainWorker](docs/TrainWorker.md)
+ - [TrainrunListResponse](docs/TrainrunListResponse.md)
  - [TryLoginEmailResponse](docs/TryLoginEmailResponse.md)
  - [TryloginemailresponseResponse](docs/TryloginemailresponseResponse.md)
  - [UXInstance](docs/UXInstance.md)

anyscale/client/openapi_client/__init__.py

@@ -31,6 +31,7 @@ from openapi_client.exceptions import ApiException
 from openapi_client.models.aws_credentials import AWSCredentials
 from openapi_client.models.aws_memory_db_cluster_config import AWSMemoryDBClusterConfig
 from openapi_client.models.access_config import AccessConfig
+from openapi_client.models.actor_status import ActorStatus
 from openapi_client.models.admin_create_user import AdminCreateUser
 from openapi_client.models.admin_created_user import AdminCreatedUser
 from openapi_client.models.admincreateduser_list_response import AdmincreateduserListResponse
@@ -216,7 +217,7 @@ from openapi_client.models.create_otp_return_api_model import CreateOTPReturnApi
 from openapi_client.models.create_organization_configuration import CreateOrganizationConfiguration
 from openapi_client.models.create_organization_invitation import CreateOrganizationInvitation
 from openapi_client.models.create_production_job_config import CreateProductionJobConfig
-from openapi_client.models.create_resource_alert import CreateResourceAlert
+from openapi_client.models.create_resource_notification import CreateResourceNotification
 from openapi_client.models.create_resource_quota import CreateResourceQuota
 from openapi_client.models.create_schedule import CreateSchedule
 from openapi_client.models.create_session_from_snapshot_options import CreateSessionFromSnapshotOptions
@@ -234,7 +235,6 @@ from openapi_client.models.createotpreturnapimodel_response import Createotpretu
 from openapi_client.models.createsessionresponse_response import CreatesessionresponseResponse
 from openapi_client.models.credit_card_information import CreditCardInformation
 from openapi_client.models.customer_alert_status import CustomerAlertStatus
-from openapi_client.models.customer_billing_type import CustomerBillingType
 from openapi_client.models.dataplane_services import DataplaneServices
 from openapi_client.models.dataset import Dataset
 from openapi_client.models.dataset_list_response import DatasetListResponse
@@ -360,6 +360,7 @@ from openapi_client.models.invoice_list_response import InvoiceListResponse
 from openapi_client.models.invoice_status import InvoiceStatus
 from openapi_client.models.invoices_query import InvoicesQuery
 from openapi_client.models.job_access import JobAccess
+from openapi_client.models.job_details import JobDetails
 from openapi_client.models.job_queue import JobQueue
 from openapi_client.models.job_queue_config import JobQueueConfig
 from openapi_client.models.job_queue_execution_mode import JobQueueExecutionMode
@@ -384,7 +385,7 @@ 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_resource_alerts_query import ListResourceAlertsQuery
+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
@@ -411,6 +412,7 @@ from openapi_client.models.longrunningworkload_list_response import Longrunningw
 from openapi_client.models.machine_allocation_state import MachineAllocationState
 from openapi_client.models.machine_connection_state import MachineConnectionState
 from openapi_client.models.machine_info import MachineInfo
+from openapi_client.models.machine_launch_failure import MachineLaunchFailure
 from openapi_client.models.machine_pool import MachinePool
 from openapi_client.models.machine_state_info import MachineStateInfo
 from openapi_client.models.metronome_customer_info_model import MetronomeCustomerInfoModel
@@ -520,17 +522,19 @@ from openapi_client.models.request_state_info import RequestStateInfo
 from openapi_client.models.requestemailmagiclinkresponse_response import RequestemailmagiclinkresponseResponse
 from openapi_client.models.requestotpreturnapimodel_response import RequestotpreturnapimodelResponse
 from openapi_client.models.reset_password_params import ResetPasswordParams
-from openapi_client.models.resource_alert import ResourceAlert
 from openapi_client.models.resource_alert_event_type import ResourceAlertEventType
+from openapi_client.models.resource_notification import ResourceNotification
 from openapi_client.models.resource_quota import ResourceQuota
 from openapi_client.models.resource_quota_status import ResourceQuotaStatus
-from openapi_client.models.resourcealert_list_response import ResourcealertListResponse
-from openapi_client.models.resourcealert_response import ResourcealertResponse
+from openapi_client.models.resourcenotification_list_response import ResourcenotificationListResponse
+from openapi_client.models.resourcenotification_response import ResourcenotificationResponse
 from openapi_client.models.resourcequota_list_response import ResourcequotaListResponse
 from openapi_client.models.resourcequota_response import ResourcequotaResponse
 from openapi_client.models.resources import Resources
 from openapi_client.models.rollback_service_model import RollbackServiceModel
 from openapi_client.models.rollout_strategy import RolloutStrategy
+from openapi_client.models.run_attempt_status import RunAttemptStatus
+from openapi_client.models.run_status import RunStatus
 from openapi_client.models.s3_download_location import S3DownloadLocation
 from openapi_client.models.sso_login_info import SSOLoginInfo
 from openapi_client.models.supportedbaseimagesenum import SUPPORTEDBASEIMAGESENUM
@@ -616,6 +620,11 @@ from openapi_client.models.timestamped_logs_output import TimestampedLogsOutput
 from openapi_client.models.timestampedlogsoutput_response import TimestampedlogsoutputResponse
 from openapi_client.models.tool import Tool
 from openapi_client.models.tracing_config import TracingConfig
+from openapi_client.models.train_resources import TrainResources
+from openapi_client.models.train_run import TrainRun
+from openapi_client.models.train_run_attempt import TrainRunAttempt
+from openapi_client.models.train_worker import TrainWorker
+from openapi_client.models.trainrun_list_response import TrainrunListResponse
 from openapi_client.models.try_login_email_response import TryLoginEmailResponse
 from openapi_client.models.tryloginemailresponse_response import TryloginemailresponseResponse
 from openapi_client.models.ux_instance import UXInstance