anyscale 0.24.28__tar.gz → 0.24.30__tar.gz

{anyscale-0.24.28 → anyscale-0.24.30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: anyscale
-Version: 0.24.28
+Version: 0.24.30
 Summary: Command Line Interface for Anyscale
 Author: Anyscale Inc.
 License: AS License

{anyscale-0.24.28 → anyscale-0.24.30}/anyscale/_private/anyscale_client/anyscale_client.py

@@ -413,7 +413,10 @@ class AnyscaleClient(AnyscaleClientInterface):
             raise e from None
 
     def get_compute_config_id(
-        self, compute_config_name: Optional[str] = None
+        self,
+        compute_config_name: Optional[str] = None,
+        *,
+        include_archived: bool = False,
     ) -> Optional[str]:
         if compute_config_name is not None:
             name, version = parse_cluster_compute_name_version(compute_config_name)
@@ -425,7 +428,9 @@ class AnyscaleClient(AnyscaleClientInterface):
                     orgwide=True,
                     name={"equals": name},
                     include_anonymous=True,
-                    archive_status=ArchiveStatus.NOT_ARCHIVED,
+                    archive_status=ArchiveStatus.ALL
+                    if include_archived
+                    else ArchiveStatus.NOT_ARCHIVED,
                     version=version,
                 )
             ).results

{anyscale-0.24.28 → anyscale-0.24.30}/anyscale/_private/anyscale_client/common.py

@@ -122,7 +122,7 @@ class AnyscaleClientInterface(ABC):
 
     @abstractmethod
     def get_compute_config_id(
-        self, compute_config_name: Optional[str] = None
+        self, compute_config_name: Optional[str] = None, *, include_archived=False
     ) -> Optional[str]:
         """Get the compute config ID for the provided name.
 

{anyscale-0.24.28 → anyscale-0.24.30}/anyscale/_private/anyscale_client/fake_anyscale_client.py

@@ -240,10 +240,16 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
     def get_compute_config(
         self, compute_config_id: str
     ) -> Optional[DecoratedComputeTemplate]:
-        return self._compute_configs.get(compute_config_id, None)
+        if compute_config_id in self._compute_configs:
+            return self._compute_configs[compute_config_id]
+
+        if compute_config_id in self._archived_compute_configs:
+            return self._archived_compute_configs[compute_config_id]
+
+        return None
 
     def get_compute_config_id(
-        self, compute_config_name: Optional[str] = None
+        self, compute_config_name: Optional[str] = None, *, include_archived=False
     ) -> Optional[str]:
         if compute_config_name is not None:
             name, version = parse_cluster_compute_name_version(compute_config_name)
@@ -253,7 +259,14 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
             if version is None:
                 version = len(self._compute_config_name_to_ids[name])
 
-            return self._compute_config_name_to_ids[name][version - 1]
+            compute_config_id = self._compute_config_name_to_ids[name][version - 1]
+            if (
+                not include_archived
+                and compute_config_id in self._archived_compute_configs
+            ):
+                return None
+
+            return compute_config_id
 
         workspace_cluster = self.get_current_workspace_cluster()
         if workspace_cluster is not None:

{anyscale-0.24.28 → anyscale-0.24.30}/anyscale/_private/docgen/generator.py

@@ -283,8 +283,9 @@ class MarkdownGenerator:
         elif isinstance(t, ModelEnumType):
             md += "#### Values\n\n"
             for value in t.__members__:
-                docstring = t.__docstrings__[value]
-                md += f" - **`{value}`**: {docstring}\n"
+                if not str(value).startswith("_"):
+                    docstring = t.__docstrings__[value]
+                    md += f" - **`{value}`**: {docstring}\n"
             md += "\n"
 
         return md

{anyscale-0.24.28 → anyscale-0.24.30}/anyscale/_private/models/model_base.py

@@ -1,7 +1,7 @@
 from dataclasses import asdict, fields
 from enum import Enum, EnumMeta
 import inspect
-from typing import Any, Dict, Iterable, Tuple, TypeVar
+from typing import Any, Dict, Iterable, Tuple, TypeVar, Union
 
 import yaml
 
@@ -25,7 +25,9 @@ class ModelEnumType(EnumMeta):
                 "Must be a Dict[ModelEnum, str] documenting all enum values"
             )
 
-        missing = {v.value for v in cls} - set(cls.__docstrings__.keys())
+        missing = {v.value for v in cls if not v.name.startswith("_")} - set(
+            cls.__docstrings__.keys()
+        )
         if missing:
             raise ValueError(
                 f"ModelEnum '{cls.__name__}.__docstrings__' is missing docstrings for values: {sorted(missing)}"
@@ -41,6 +43,16 @@ class ModelEnum(str, Enum, metaclass=ModelEnumType):
     def __str__(self):
         return self.name
 
+    @classmethod
+    def validate(cls, value: Union[str, "ModelEnum"]) -> "ModelEnum":
+        allowed_values = [str(s) for s in cls.__members__ if not str(s).startswith("_")]
+        if value.upper() in allowed_values:
+            return cls(value.upper())
+        else:
+            raise ValueError(
+                f"'{value.upper()}' is not a valid {cls.__name__}. Allowed values are {allowed_values}"
+            )
+
 
 class ModelBaseType(type):
     pass

{anyscale-0.24.28 → anyscale-0.24.30}/anyscale/client/README.md

@@ -85,6 +85,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**archive_compute_template_api_v2_compute_templates_compute_template_id_archive_post**](docs/DefaultApi.md#archive_compute_template_api_v2_compute_templates_compute_template_id_archive_post) | **POST** /api/v2/compute_templates/{compute_template_id}/archive | Archive Compute Template
 *DefaultApi* | [**archive_job_api_v2_decorated_ha_jobs_production_job_id_archive_post**](docs/DefaultApi.md#archive_job_api_v2_decorated_ha_jobs_production_job_id_archive_post) | **POST** /api/v2/decorated_ha_jobs/{production_job_id}/archive | Archive Job
 *DefaultApi* | [**attach_machine_pool_to_cloud_api_v2_machine_pools_attach_post**](docs/DefaultApi.md#attach_machine_pool_to_cloud_api_v2_machine_pools_attach_post) | **POST** /api/v2/machine_pools/attach | Attach Machine Pool To Cloud
+*DefaultApi* | [**ban_organization_api_v2_organizations_organization_id_ban_put**](docs/DefaultApi.md#ban_organization_api_v2_organizations_organization_id_ban_put) | **PUT** /api/v2/organizations/{organization_id}/ban | Ban Organization
 *DefaultApi* | [**batch_create_invitations_api_v2_organization_invitations_batch_create_post**](docs/DefaultApi.md#batch_create_invitations_api_v2_organization_invitations_batch_create_post) | **POST** /api/v2/organization_invitations/batch_create | Batch Create Invitations
 *DefaultApi* | [**batch_get_job_api_v2_decorated_ha_jobs_batch_get_post**](docs/DefaultApi.md#batch_get_job_api_v2_decorated_ha_jobs_batch_get_post) | **POST** /api/v2/decorated_ha_jobs/batch_get | Batch Get Job
 *DefaultApi* | [**begin_cluster_startup_api_v2_cluster_startups_post**](docs/DefaultApi.md#begin_cluster_startup_api_v2_cluster_startups_post) | **POST** /api/v2/cluster_startups/ | Begin Cluster Startup

{anyscale-0.24.28 → anyscale-0.24.30}/anyscale/client/openapi_client/api/default_api.py

@@ -1219,6 +1219,118 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def ban_organization_api_v2_organizations_organization_id_ban_put(self, organization_id, **kwargs):  # noqa: E501
+        """Ban Organization  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.ban_organization_api_v2_organizations_organization_id_ban_put(organization_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str organization_id: (required)
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: object
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.ban_organization_api_v2_organizations_organization_id_ban_put_with_http_info(organization_id, **kwargs)  # noqa: E501
+
+    def ban_organization_api_v2_organizations_organization_id_ban_put_with_http_info(self, organization_id, **kwargs):  # noqa: E501
+        """Ban Organization  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.ban_organization_api_v2_organizations_organization_id_ban_put_with_http_info(organization_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str organization_id: (required)
+        :param _return_http_data_only: response data without head status code
+                                       and headers
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: tuple(object, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'organization_id'
+        ]
+        all_params.extend(
+            [
+                'async_req',
+                '_return_http_data_only',
+                '_preload_content',
+                '_request_timeout'
+            ]
+        )
+
+        for key, val in six.iteritems(local_var_params['kwargs']):
+            if key not in all_params:
+                raise ApiTypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method ban_organization_api_v2_organizations_organization_id_ban_put" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'organization_id' is set
+        if self.api_client.client_side_validation and ('organization_id' not in local_var_params or  # noqa: E501
+                                                        local_var_params['organization_id'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `organization_id` when calling `ban_organization_api_v2_organizations_organization_id_ban_put`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'organization_id' in local_var_params:
+            path_params['organization_id'] = local_var_params['organization_id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/organizations/{organization_id}/ban', 'PUT',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='object',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=local_var_params.get('async_req'),
+            _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
+            _preload_content=local_var_params.get('_preload_content', True),
+            _request_timeout=local_var_params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def batch_create_invitations_api_v2_organization_invitations_batch_create_post(self, create_organization_invitation, **kwargs):  # noqa: E501
         """Batch Create Invitations  # noqa: E501
 

{anyscale-0.24.28 → anyscale-0.24.30}/anyscale/client/openapi_client/models/company_size.py

@@ -29,11 +29,11 @@ class CompanySize(object):
     allowed enum values
     """
     _1_50 = "1 - 50"
-    _50_200 = "50 - 200"
-    _200_500 = "200 - 500"
+    _51_200 = "51 - 200"
+    _201_500 = "201 - 500"
     _500 = "> 500"
 
-    allowable_values = [_1_50, _50_200, _200_500, _500]  # noqa: E501
+    allowable_values = [_1_50, _51_200, _201_500, _500]  # noqa: E501
 
     """
     Attributes:

{anyscale-0.24.28 → anyscale-0.24.30}/anyscale/client/openapi_client/models/organization.py

@@ -45,7 +45,8 @@ class Organization(object):
         'endpoints_user_id': 'str',
         'endpoints_stripe_customer_id': 'str',
         'endpoints_metronome_customer_id': 'str',
-        'endpoints_is_free_credit_used_up': 'bool'
+        'endpoints_is_free_credit_used_up': 'bool',
+        'metronome_customer_id': 'str'
     }
 
     attribute_map = {
@@ -61,10 +62,11 @@ class Organization(object):
         'endpoints_user_id': 'endpoints_user_id',
         'endpoints_stripe_customer_id': 'endpoints_stripe_customer_id',
         'endpoints_metronome_customer_id': 'endpoints_metronome_customer_id',
-        'endpoints_is_free_credit_used_up': 'endpoints_is_free_credit_used_up'
+        'endpoints_is_free_credit_used_up': 'endpoints_is_free_credit_used_up',
+        'metronome_customer_id': 'metronome_customer_id'
     }
 
-    def __init__(self, id=None, name=None, public_identifier=None, default_cloud_id=None, sso_config_id=None, sso_required=False, is_general_platform_enabled=None, is_private_endpoints_enabled=None, is_usage_blocked=None, endpoints_user_id=None, endpoints_stripe_customer_id=None, endpoints_metronome_customer_id=None, endpoints_is_free_credit_used_up=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, id=None, name=None, public_identifier=None, default_cloud_id=None, sso_config_id=None, sso_required=False, is_general_platform_enabled=None, is_private_endpoints_enabled=None, is_usage_blocked=None, endpoints_user_id=None, endpoints_stripe_customer_id=None, endpoints_metronome_customer_id=None, endpoints_is_free_credit_used_up=None, metronome_customer_id=None, local_vars_configuration=None):  # noqa: E501
         """Organization - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -83,6 +85,7 @@ class Organization(object):
         self._endpoints_stripe_customer_id = None
         self._endpoints_metronome_customer_id = None
         self._endpoints_is_free_credit_used_up = None
+        self._metronome_customer_id = None
         self.discriminator = None
 
         self.id = id
@@ -105,6 +108,8 @@ class Organization(object):
             self.endpoints_metronome_customer_id = endpoints_metronome_customer_id
         if endpoints_is_free_credit_used_up is not None:
             self.endpoints_is_free_credit_used_up = endpoints_is_free_credit_used_up
+        if metronome_customer_id is not None:
+            self.metronome_customer_id = metronome_customer_id
 
     @property
     def id(self):
@@ -391,6 +396,27 @@ class Organization(object):
 
         self._endpoints_is_free_credit_used_up = endpoints_is_free_credit_used_up
 
+    @property
+    def metronome_customer_id(self):
+        """Gets the metronome_customer_id of this Organization.  # noqa: E501
+
+
+        :return: The metronome_customer_id of this Organization.  # noqa: E501
+        :rtype: str
+        """
+        return self._metronome_customer_id
+
+    @metronome_customer_id.setter
+    def metronome_customer_id(self, metronome_customer_id):
+        """Sets the metronome_customer_id of this Organization.
+
+
+        :param metronome_customer_id: The metronome_customer_id of this Organization.  # noqa: E501
+        :type: str
+        """
+
+        self._metronome_customer_id = metronome_customer_id
+
     def to_dict(self):
         """Returns the model properties as a dict"""
         result = {}

{anyscale-0.24.28 → anyscale-0.24.30}/anyscale/commands/compute_config_commands.py

@@ -7,11 +7,6 @@ from anyscale.util import validate_non_negative_arg
 from anyscale.utils.entity_arg_utils import format_inputs_to_entity
 
 
-CLUSTER_COMPUTE_ID_DEPRECATED_ERROR = click.ClickException(
-    "'--cluster-compute-id' is fully deprecated, use '--compute-config-id' instead."
-)
-
-
 @click.group(
     "compute-config",
     short_help="Manage compute configurations on Anyscale.",
@@ -40,43 +35,6 @@ def create_compute_config(compute_config_file: IO[bytes], name: Optional[str]) -
     ComputeConfigController().create(compute_config_file, name)
 
 
-@compute_config_cli.command(
-    name="delete", help="Delete the specified compute config.", hidden=True
-)
-@click.argument("compute-config-name", type=str, required=False)
-@click.option(
-    "--name",
-    "-n",
-    help="Name of the compute config to delete.",
-    required=False,
-    type=str,
-)
-@click.option(
-    "--cluster-compute-id",
-    help="Deprecated. Use --compute-config-id instead. Id of the compute config to archive. Must be provided if a compute name is not given.",
-    required=False,
-    type=str,
-    hidden=True,
-)
-@click.option(
-    "--compute-config-id",
-    "--id",
-    help="Id of the compute config to delete. Must be provided if a compute name is not given.",
-    required=False,
-    type=str,
-)
-def delete_compute_config(
-    compute_config_name: Optional[str],  # noqa: ARG001
-    name: Optional[str],  # noqa: ARG001
-    cluster_compute_id: Optional[str],  # noqa: ARG001
-    compute_config_id: Optional[str],  # noqa: ARG001
-):
-    raise click.ClickException(
-        "'anyscale compute-config delete' is fully deprecated, "
-        "use 'anyscale compute-config archive' instead."
-    )
-
-
 @compute_config_cli.command(
     name="archive", help="Archive the specified compute config.",
 )
@@ -88,13 +46,6 @@ def delete_compute_config(
     required=False,
     type=str,
 )
-@click.option(
-    "--cluster-compute-id",
-    help="Deprecated. Use --compute-config-id instead. Id of the compute config to archive. Must be provided if a compute name is not given.",
-    required=False,
-    type=str,
-    hidden=True,
-)
 @click.option(
     "--compute-config-id",
     "--id",
@@ -105,12 +56,8 @@ def delete_compute_config(
 def archive_compute_config(
     compute_config_name: Optional[str],
     name: Optional[str],
-    cluster_compute_id: Optional[str],
     compute_config_id: Optional[str],
 ) -> None:
-    if cluster_compute_id:
-        raise CLUSTER_COMPUTE_ID_DEPRECATED_ERROR
-
     if compute_config_name is not None and name is not None:
         raise click.ClickException(
             "Please only provide one of [COMPUTE_CONFIG_NAME] or --name."
@@ -134,15 +81,6 @@ def archive_compute_config(
     default=None,
     help="List information about the compute config with this name.",
 )
-@click.option(
-    "--cluster-compute-id",
-    required=False,
-    default=None,
-    help=(
-        "Deprecated. Use --compute-config-id instead. List information about the compute config with this id."
-    ),
-    hidden=True,
-)
 @click.option(
     "--compute-config-id",
     "--id",
@@ -166,14 +104,10 @@ def archive_compute_config(
 )
 def list(  # noqa: A001
     name: Optional[str],
-    cluster_compute_id: Optional[str],
     compute_config_id: Optional[str],
     include_shared: bool,
     max_items: int,
 ):
-    if cluster_compute_id:
-        raise CLUSTER_COMPUTE_ID_DEPRECATED_ERROR
-
     ComputeConfigController().list(
         cluster_compute_name=name,
         cluster_compute_id=compute_config_id,
@@ -186,36 +120,21 @@ def list(  # noqa: A001
     name="get", help=("Get details about compute configuration."),
 )
 @click.argument("compute-config-name", required=False)
-@click.option(
-    "--cluster-compute-id",
-    required=False,
-    default=None,
-    help=(
-        "Deprecated. Use --compute-config-id instead. Get details about compute configuration by this id."
-    ),
-    hidden=True,
-)
 @click.option(
     "--compute-config-id",
     "--id",
     required=False,
     default=None,
-    help=("Get details about compute configuration by this id."),
+    help="Get details about compute configuration by this id.",
 )
 @click.option(
-    "--include-archived",
-    is_flag=True,
-    help=("Include archived compute configurations."),
+    "--include-archived", is_flag=True, help="Include archived compute configurations.",
 )
 def get(
     compute_config_name: Optional[str],
-    cluster_compute_id: Optional[str],
     compute_config_id: Optional[str],
     include_archived: bool,
 ):
-    if cluster_compute_id:
-        raise CLUSTER_COMPUTE_ID_DEPRECATED_ERROR
-
     ComputeConfigController().get(
         cluster_compute_name=compute_config_name,
         cluster_compute_id=compute_config_id,

{anyscale-0.24.28 → anyscale-0.24.30}/anyscale/commands/job_commands.py

@@ -9,10 +9,9 @@ import yaml
 
 import anyscale
 from anyscale.cli_logger import BlockLogger
-from anyscale.client.openapi_client import HaJobStates
 from anyscale.commands.util import convert_kv_strings_to_dict
 from anyscale.controllers.job_controller import JobController
-from anyscale.job.models import JobConfig, JobStatus
+from anyscale.job.models import JobConfig, JobState, JobStatus
 from anyscale.util import validate_non_negative_arg
 
 
@@ -42,6 +41,7 @@ def job_cli() -> None:
     "-f",
     "--follow",
     required=False,
+    hidden=True,
     default=False,
     type=bool,
     is_flag=True,
@@ -204,8 +204,11 @@ and override the entrypoint with `python main.py`.
             )
         if config_file is not None and not pathlib.Path(config_file).is_file():
             raise click.ClickException(f"Job config file '{config_file}' not found.")
+
         if follow:
-            log.warning("`--follow` is deprecated, use `--wait` instead.")
+            raise click.ClickException(
+                "`--follow` is deprecated, use `--wait` instead."
+            )
 
         args = {}
         if len(entrypoint) > 0:
@@ -259,19 +262,15 @@ and override the entrypoint with `python main.py`.
         log.info(f"Submitting job with config {config}.")
         job_id = anyscale.job.submit(config)
 
-    if follow or wait:
-        job_controller.log.info(
+    if wait:
+        log.info(
             "Waiting for the job to run. Interrupting this command will not cancel the job."
         )
     else:
-        job_controller.log.info(
-            "Use `--wait` to wait for the job to run and stream logs."
-        )
+        log.info("Use `--wait` to wait for the job to run and stream logs.")
 
-    if follow:
-        job_controller.logs(job_id, should_follow=follow)
     if wait:
-        job_controller.wait(job_id=job_id)
+        anyscale.job.wait(job_id=job_id)
 
 
 @job_cli.command(name="list", help="Display information about existing jobs.")
@@ -412,24 +411,25 @@ def logs(
     "--state",
     "-s",
     required=False,
-    default=HaJobStates.SUCCESS,
+    default=JobState.SUCCEEDED,
     help="The state to wait for this job to enter",
 )
 @click.option(
+    "--timeout-s",
     "--timeout",
     "-t",
     required=False,
-    default=None,
+    default=1800,
     type=float,
     help="The timeout in seconds after which this command will exit.",
 )
 def wait(
     job_id: Optional[str],
     name: Optional[str],
-    state: str = HaJobStates.SUCCESS,
-    timeout=None,
+    state: str = JobState.SUCCEEDED,
+    timeout_s=None,
 ) -> None:
-    """Wait for a job to enter a specific state (default: SUCCESS).
+    """Wait for a job to enter a specific state (default: SUCCEEDED).
 
     To specify the job by name, use the --name flag. To specify the job by id, use the --job-id flag.
 
@@ -437,13 +437,13 @@ def wait(
 
     If the job reaches a terminal state other than the target state, the command will exit with an error.
 
-    If the command reaches the timeout, the command will exit with an error.
+    If the command reaches the timeout, the command will exit with an error but job execution will continue.
     """
-    state = state.upper()
-    job_controller = JobController()
-    job_id = job_controller.wait(
-        job_id=job_id, job_name=name, target_state=state, timeout_secs=timeout,
-    )
+    try:
+        state = JobState.validate(state)
+    except ValueError as e:
+        raise click.ClickException(str(e))
+    anyscale.job.wait(name=name, job_id=job_id, state=state, timeout_s=timeout_s)
 
 
 @job_cli.command(

{anyscale-0.24.28 → anyscale-0.24.30}/anyscale/commands/service_commands.py

@@ -340,6 +340,11 @@ def status(name: Optional[str], config_file: Optional[str], json: bool, verbose:
     help="Timeout to wait for the service to reach the target state. Defaults to 600s (10min).",
 )
 def wait(name: Optional[str], config_file: Optional[str], state: str, timeout_s: int):
+    try:
+        state = ServiceState.validate(state)
+    except ValueError as e:
+        raise click.ClickException(str(e))
+
     if name is not None and config_file is not None:
         raise click.ClickException(
             "Only one of '--name' and '--config-file' can be provided."