anyscale 0.26.9__py3-none-any.whl → 0.26.10__py3-none-any.whl

anyscale/anyscale_halo/halo_notebook.py

@@ -0,0 +1,122 @@
+from __future__ import absolute_import, print_function, unicode_literals
+
+import sys
+import threading
+
+import anyscale.anyscale_halo.cursor as cursor
+
+from anyscale.anyscale_halo import Halo
+from anyscale.anyscale_halo._utils import colored_frame, decode_utf_8_text
+
+
+class HaloNotebook(Halo):
+    def __init__(
+        self,
+        text="",
+        color="cyan",
+        text_color=None,
+        spinner=None,
+        placement="left",
+        animation=None,
+        interval=-1,
+        enabled=True,
+        stream=sys.stdout,
+    ):
+        super(HaloNotebook, self).__init__(
+            text=text,
+            color=color,
+            text_color=text_color,
+            spinner=spinner,
+            placement=placement,
+            animation=animation,
+            interval=interval,
+            enabled=enabled,
+            stream=stream,
+        )
+        self.output = self._make_output_widget()
+
+    def _make_output_widget(self):
+        from ipywidgets.widgets import Output
+
+        return Output()
+
+    # TODO: using property and setter
+    def _output(self, text=""):
+        return ({"name": "stdout", "output_type": "stream", "text": text},)
+
+    def clear(self):
+        if not self.enabled:
+            return self
+
+        with self.output:
+            self.output.outputs += self._output("\r")
+            self.output.outputs += self._output(self.CLEAR_LINE)
+
+        self.output.outputs = self._output()
+        return self
+
+    def _render_frame(self):
+        frame = self.frame()
+        output = "\r{}".format(frame)
+        with self.output:
+            self.output.outputs += self._output(output)
+
+    def start(self, text=None):
+        if text is not None:
+            self.text = text
+
+        if not self.enabled or self._spinner_id is not None:
+            return self
+
+        if self._stream.isatty():
+            cursor.hide()
+
+        self.output = self._make_output_widget()
+        from IPython.display import display
+
+        display(self.output)
+        self._stop_spinner = threading.Event()
+        self._spinner_thread = threading.Thread(target=self.render)
+        self._spinner_thread.setDaemon(True)
+        self._render_frame()
+        self._spinner_id = self._spinner_thread.name
+        self._spinner_thread.start()
+
+        return self
+
+    def stop_and_persist(self, symbol=" ", text=None):
+        """Stops the spinner and persists the final frame to be shown.
+        Parameters
+        ----------
+        symbol : str, optional
+            Symbol to be shown in final frame
+        text: str, optional
+            Text to be shown in final frame
+
+        Returns
+        -------
+        self
+        """
+        if not self.enabled:
+            return self
+
+        symbol = decode_utf_8_text(symbol)
+
+        if text is not None:
+            text = decode_utf_8_text(text)
+        else:
+            text = self._text["original"]
+
+        text = text.strip()
+
+        if self._text_color:
+            text = colored_frame(text, self._text_color)
+
+        self.stop()
+
+        output = "\r{} {}\n".format(
+            *[(text, symbol) if self._placement == "right" else (symbol, text)][0]
+        )
+
+        with self.output:
+            self.output.outputs = self._output(output)

anyscale/cli_logger.py

@@ -8,8 +8,8 @@ from typing import List, Optional
 from botocore.exceptions import ClientError
 import click
 import colorama
-from halo import Halo
 
+from anyscale.anyscale_halo import Halo
 from anyscale.client.openapi_client.models import (
     CloudAnalyticsEventCloudProviderError,
     CloudAnalyticsEventCloudResource,

anyscale/client/README.md

@@ -115,7 +115,6 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**create_configuration_script_api_v2_organization_configurations_create_helper_get**](docs/DefaultApi.md#create_configuration_script_api_v2_organization_configurations_create_helper_get) | **GET** /api/v2/organization_configurations/create_helper | Create Configuration Script
 *DefaultApi* | [**create_connect_session_api_v2_sessions_create_connect_session_post**](docs/DefaultApi.md#create_connect_session_api_v2_sessions_create_connect_session_post) | **POST** /api/v2/sessions/create_connect_session | Create Connect Session
 *DefaultApi* | [**create_dataset_upload_api_v2_datasets_upload_post**](docs/DefaultApi.md#create_dataset_upload_api_v2_datasets_upload_post) | **POST** /api/v2/datasets/upload | Create Dataset Upload
-*DefaultApi* | [**create_endpoint_user_for_org_api_v2_organizations_organization_id_create_endpoint_user_post**](docs/DefaultApi.md#create_endpoint_user_for_org_api_v2_organizations_organization_id_create_endpoint_user_post) | **POST** /api/v2/organizations/{organization_id}/create_endpoint_user | Create Endpoint User For Org
 *DefaultApi* | [**create_instance_usage_budget_api_v2_instance_usage_budgets_post**](docs/DefaultApi.md#create_instance_usage_budget_api_v2_instance_usage_budgets_post) | **POST** /api/v2/instance_usage_budgets/ | Create Instance Usage Budget
 *DefaultApi* | [**create_invitation_api_v2_organization_invitations_post**](docs/DefaultApi.md#create_invitation_api_v2_organization_invitations_post) | **POST** /api/v2/organization_invitations/ | Create Invitation
 *DefaultApi* | [**create_job_api_v2_decorated_ha_jobs_create_post**](docs/DefaultApi.md#create_job_api_v2_decorated_ha_jobs_create_post) | **POST** /api/v2/decorated_ha_jobs/create | Create Job
@@ -157,7 +156,6 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**describe_session_api_v2_sessions_session_id_describe_get**](docs/DefaultApi.md#describe_session_api_v2_sessions_session_id_describe_get) | **GET** /api/v2/sessions/{session_id}/describe | Describe Session
 *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* | [**dismiss_user_card_api_v2_onboarding_cards_card_id_dismiss_post**](docs/DefaultApi.md#dismiss_user_card_api_v2_onboarding_cards_card_id_dismiss_post) | **POST** /api/v2/onboarding_cards/{card_id}/dismiss | Dismiss User Card
 *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
 *DefaultApi* | [**download_aggregated_instance_usage_csv_internal_api_v2_aggregated_instance_usage_download_csv_internal_get**](docs/DefaultApi.md#download_aggregated_instance_usage_csv_internal_api_v2_aggregated_instance_usage_download_csv_internal_get) | **GET** /api/v2/aggregated_instance_usage/download_csv_internal | Download Aggregated Instance Usage Csv Internal
 *DefaultApi* | [**echo_ip_api_v2_health_echo_ip_get**](docs/DefaultApi.md#echo_ip_api_v2_health_echo_ip_get) | **GET** /api/v2/health/echo_ip | Echo Ip
@@ -214,7 +212,6 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**get_events_api_v2_sessions_session_id_events_get**](docs/DefaultApi.md#get_events_api_v2_sessions_session_id_events_get) | **GET** /api/v2/sessions/{session_id}/events | Get Events
 *DefaultApi* | [**get_execution_logs_api_v2_session_commands_session_command_id_execution_logs_get**](docs/DefaultApi.md#get_execution_logs_api_v2_session_commands_session_command_id_execution_logs_get) | **GET** /api/v2/session_commands/{session_command_id}/execution_logs | Get Execution Logs
 *DefaultApi* | [**get_execution_logs_archived_api_v2_session_commands_session_command_id_execution_logs_archived_get**](docs/DefaultApi.md#get_execution_logs_archived_api_v2_session_commands_session_command_id_execution_logs_archived_get) | **GET** /api/v2/session_commands/{session_command_id}/execution_logs_archived | Get Execution Logs Archived
-*DefaultApi* | [**get_feature_compatibility_api_v2_product_features_get**](docs/DefaultApi.md#get_feature_compatibility_api_v2_product_features_get) | **GET** /api/v2/product_features/ | Get Feature Compatibility
 *DefaultApi* | [**get_instance_usage_budget_api_v2_instance_usage_budgets_instance_usage_budget_id_get**](docs/DefaultApi.md#get_instance_usage_budget_api_v2_instance_usage_budgets_instance_usage_budget_id_get) | **GET** /api/v2/instance_usage_budgets/{instance_usage_budget_id} | Get Instance Usage Budget
 *DefaultApi* | [**get_invitation_api_v2_organization_invitations_invitation_id_get**](docs/DefaultApi.md#get_invitation_api_v2_organization_invitations_invitation_id_get) | **GET** /api/v2/organization_invitations/{invitation_id} | Get Invitation
 *DefaultApi* | [**get_job_api_v2_decorated_ha_jobs_production_job_id_get**](docs/DefaultApi.md#get_job_api_v2_decorated_ha_jobs_production_job_id_get) | **GET** /api/v2/decorated_ha_jobs/{production_job_id} | Get Job
@@ -356,7 +353,6 @@ Class | Method | HTTP request | Description
 *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
 *DefaultApi* | [**set_resource_quota_status_api_v2_resource_quotas_resource_quota_id_status_patch**](docs/DefaultApi.md#set_resource_quota_status_api_v2_resource_quotas_resource_quota_id_status_patch) | **PATCH** /api/v2/resource_quotas/{resource_quota_id}/status | Set Resource Quota Status
 *DefaultApi* | [**show_one_time_password_source_api_v2_users_show_otp_source_post**](docs/DefaultApi.md#show_one_time_password_source_api_v2_users_show_otp_source_post) | **POST** /api/v2/users/show_otp_source | Show One Time Password Source
 *DefaultApi* | [**sso_login_info_api_v2_users_sso_login_info_get**](docs/DefaultApi.md#sso_login_info_api_v2_users_sso_login_info_get) | **GET** /api/v2/users/sso_login_info | Sso Login Info
@@ -371,7 +367,6 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**trigger_cron_job_api_v2_experimental_cron_jobs_cron_job_id_trigger_post**](docs/DefaultApi.md#trigger_cron_job_api_v2_experimental_cron_jobs_cron_job_id_trigger_post) | **POST** /api/v2/experimental_cron_jobs/{cron_job_id}/trigger | Trigger Cron Job
 *DefaultApi* | [**try_claim_cloud_api_v2_aioa_cloud_waitlist_claim_cloud_post**](docs/DefaultApi.md#try_claim_cloud_api_v2_aioa_cloud_waitlist_claim_cloud_post) | **POST** /api/v2/aioa_cloud_waitlist/claim_cloud | Try Claim Cloud
 *DefaultApi* | [**try_login_api_v2_users_try_login_post**](docs/DefaultApi.md#try_login_api_v2_users_try_login_post) | **POST** /api/v2/users/try-login | Try Login
-*DefaultApi* | [**undismiss_user_card_api_v2_onboarding_cards_card_id_undismiss_post**](docs/DefaultApi.md#undismiss_user_card_api_v2_onboarding_cards_card_id_undismiss_post) | **POST** /api/v2/onboarding_cards/{card_id}/undismiss | Undismiss User Card
 *DefaultApi* | [**update_cloud_auto_add_user_api_v2_clouds_cloud_id_auto_add_user_put**](docs/DefaultApi.md#update_cloud_auto_add_user_api_v2_clouds_cloud_id_auto_add_user_put) | **PUT** /api/v2/clouds/{cloud_id}/auto_add_user | Update Cloud Auto Add User
 *DefaultApi* | [**update_cloud_config_api_v2_clouds_cloud_id_config_put**](docs/DefaultApi.md#update_cloud_config_api_v2_clouds_cloud_id_config_put) | **PUT** /api/v2/clouds/{cloud_id}/config | Update Cloud Config
 *DefaultApi* | [**update_cloud_deployment_config_api_v2_clouds_cloud_id_deployment_cloud_deployment_id_config_put**](docs/DefaultApi.md#update_cloud_deployment_config_api_v2_clouds_cloud_id_deployment_cloud_deployment_id_config_put) | **PUT** /api/v2/clouds/{cloud_id}/deployment/{cloud_deployment_id}/config | Update Cloud Deployment Config
@@ -453,9 +448,6 @@ Class | Method | HTTP request | Description
  - [BuildResponse](docs/BuildResponse.md)
  - [BuildStatus](docs/BuildStatus.md)
  - [BuildlogresponseResponse](docs/BuildlogresponseResponse.md)
- - [Card](docs/Card.md)
- - [CardId](docs/CardId.md)
- - [CardListResponse](docs/CardListResponse.md)
  - [ChangePasswordParams](docs/ChangePasswordParams.md)
  - [CleanupLeakedGrafanaDashboardResponse](docs/CleanupLeakedGrafanaDashboardResponse.md)
  - [CleanupleakedgrafanadashboardresponseResponse](docs/CleanupleakedgrafanadashboardresponseResponse.md)
@@ -517,7 +509,6 @@ Class | Method | HTTP request | Description
  - [ClusterEvent](docs/ClusterEvent.md)
  - [ClusterEventSource](docs/ClusterEventSource.md)
  - [ClusterEventsOutput](docs/ClusterEventsOutput.md)
- - [ClusterFeatures](docs/ClusterFeatures.md)
  - [ClusterManagementStackVersions](docs/ClusterManagementStackVersions.md)
  - [ClusterSize](docs/ClusterSize.md)
  - [ClusterState](docs/ClusterState.md)
@@ -526,7 +517,6 @@ Class | Method | HTTP request | Description
  - [ClusterauthresponseResponse](docs/ClusterauthresponseResponse.md)
  - [ClustereventListResponse](docs/ClustereventListResponse.md)
  - [ClustereventsoutputResponse](docs/ClustereventsoutputResponse.md)
- - [ClusterfeaturesResponse](docs/ClusterfeaturesResponse.md)
  - [ComputeNodeType](docs/ComputeNodeType.md)
  - [ComputeStack](docs/ComputeStack.md)
  - [ComputeTemplate](docs/ComputeTemplate.md)
@@ -654,7 +644,6 @@ Class | Method | HTTP request | Description
  - [DescribesystemworkloadresponseResponse](docs/DescribesystemworkloadresponseResponse.md)
  - [DetachMachinePoolFromCloudRequest](docs/DetachMachinePoolFromCloudRequest.md)
  - [DetachmachinepoolfromcloudresponseResponse](docs/DetachmachinepoolfromcloudresponseResponse.md)
- - [DismissalType](docs/DismissalType.md)
  - [EditableCloudResource](docs/EditableCloudResource.md)
  - [EditableCloudResourceGCP](docs/EditableCloudResourceGCP.md)
  - [Error](docs/Error.md)
@@ -670,7 +659,6 @@ Class | Method | HTTP request | Description
  - [ExternalServiceStatusResponse](docs/ExternalServiceStatusResponse.md)
  - [ExternalTerminalCommand](docs/ExternalTerminalCommand.md)
  - [ExternalservicestatusresponseResponse](docs/ExternalservicestatusresponseResponse.md)
- - [FeatureCompatibility](docs/FeatureCompatibility.md)
  - [FeatureFlagResponse](docs/FeatureFlagResponse.md)
  - [FeatureflagresponseResponse](docs/FeatureflagresponseResponse.md)
  - [FineTuneType](docs/FineTuneType.md)
@@ -784,7 +772,6 @@ Class | Method | HTTP request | Description
  - [NotificationChannelEmailConfig](docs/NotificationChannelEmailConfig.md)
  - [NotificationChannelSlackConfig](docs/NotificationChannelSlackConfig.md)
  - [NotificationChannelWebhookConfig](docs/NotificationChannelWebhookConfig.md)
- - [OnboardingUserCardsQuery](docs/OnboardingUserCardsQuery.md)
  - [OperatorMetricId](docs/OperatorMetricId.md)
  - [OperatorMetrics](docs/OperatorMetrics.md)
  - [Organization](docs/Organization.md)
@@ -990,7 +977,6 @@ Class | Method | HTTP request | Description
  - [UtmFields](docs/UtmFields.md)
  - [ValidateOTPParamsApiModel](docs/ValidateOTPParamsApiModel.md)
  - [ValidationError](docs/ValidationError.md)
- - [Visibility](docs/Visibility.md)
  - [WaitlistStatusResponse](docs/WaitlistStatusResponse.md)
  - [WaitlistStatusType](docs/WaitlistStatusType.md)
  - [WaitliststatusresponseResponse](docs/WaitliststatusresponseResponse.md)

anyscale/client/openapi_client/__init__.py

@@ -82,9 +82,6 @@ from openapi_client.models.build_registration import BuildRegistration
 from openapi_client.models.build_response import BuildResponse
 from openapi_client.models.build_status import BuildStatus
 from openapi_client.models.buildlogresponse_response import BuildlogresponseResponse
-from openapi_client.models.card import Card
-from openapi_client.models.card_id import CardId
-from openapi_client.models.card_list_response import CardListResponse
 from openapi_client.models.change_password_params import ChangePasswordParams
 from openapi_client.models.cleanup_leaked_grafana_dashboard_response import CleanupLeakedGrafanaDashboardResponse
 from openapi_client.models.cleanupleakedgrafanadashboardresponse_response import CleanupleakedgrafanadashboardresponseResponse
@@ -146,7 +143,6 @@ from openapi_client.models.cluster_environments_query import ClusterEnvironments
 from openapi_client.models.cluster_event import ClusterEvent
 from openapi_client.models.cluster_event_source import ClusterEventSource
 from openapi_client.models.cluster_events_output import ClusterEventsOutput
-from openapi_client.models.cluster_features import ClusterFeatures
 from openapi_client.models.cluster_management_stack_versions import ClusterManagementStackVersions
 from openapi_client.models.cluster_size import ClusterSize
 from openapi_client.models.cluster_state import ClusterState
@@ -155,7 +151,6 @@ from openapi_client.models.cluster_status_details import ClusterStatusDetails
 from openapi_client.models.clusterauthresponse_response import ClusterauthresponseResponse
 from openapi_client.models.clusterevent_list_response import ClustereventListResponse
 from openapi_client.models.clustereventsoutput_response import ClustereventsoutputResponse
-from openapi_client.models.clusterfeatures_response import ClusterfeaturesResponse
 from openapi_client.models.compute_node_type import ComputeNodeType
 from openapi_client.models.compute_stack import ComputeStack
 from openapi_client.models.compute_template import ComputeTemplate
@@ -283,7 +278,6 @@ from openapi_client.models.describemachinepoolresponse_response import Describem
 from openapi_client.models.describesystemworkloadresponse_response import DescribesystemworkloadresponseResponse
 from openapi_client.models.detach_machine_pool_from_cloud_request import DetachMachinePoolFromCloudRequest
 from openapi_client.models.detachmachinepoolfromcloudresponse_response import DetachmachinepoolfromcloudresponseResponse
-from openapi_client.models.dismissal_type import DismissalType
 from openapi_client.models.editable_cloud_resource import EditableCloudResource
 from openapi_client.models.editable_cloud_resource_gcp import EditableCloudResourceGCP
 from openapi_client.models.error import Error
@@ -299,7 +293,6 @@ from openapi_client.models.external_service_status import ExternalServiceStatus
 from openapi_client.models.external_service_status_response import ExternalServiceStatusResponse
 from openapi_client.models.external_terminal_command import ExternalTerminalCommand
 from openapi_client.models.externalservicestatusresponse_response import ExternalservicestatusresponseResponse
-from openapi_client.models.feature_compatibility import FeatureCompatibility
 from openapi_client.models.feature_flag_response import FeatureFlagResponse
 from openapi_client.models.featureflagresponse_response import FeatureflagresponseResponse
 from openapi_client.models.fine_tune_type import FineTuneType
@@ -413,7 +406,6 @@ from openapi_client.models.node_type import NodeType
 from openapi_client.models.notification_channel_email_config import NotificationChannelEmailConfig
 from openapi_client.models.notification_channel_slack_config import NotificationChannelSlackConfig
 from openapi_client.models.notification_channel_webhook_config import NotificationChannelWebhookConfig
-from openapi_client.models.onboarding_user_cards_query import OnboardingUserCardsQuery
 from openapi_client.models.operator_metric_id import OperatorMetricId
 from openapi_client.models.operator_metrics import OperatorMetrics
 from openapi_client.models.organization import Organization
@@ -619,7 +611,6 @@ from openapi_client.models.userinfo_response import UserinfoResponse
 from openapi_client.models.utm_fields import UtmFields
 from openapi_client.models.validate_otp_params_api_model import ValidateOTPParamsApiModel
 from openapi_client.models.validation_error import ValidationError
-from openapi_client.models.visibility import Visibility
 from openapi_client.models.waitlist_status_response import WaitlistStatusResponse
 from openapi_client.models.waitlist_status_type import WaitlistStatusType
 from openapi_client.models.waitliststatusresponse_response import WaitliststatusresponseResponse