anyscale 0.24.27__tar.gz → 0.24.29__tar.gz

{anyscale-0.24.27 → anyscale-0.24.29}/PKG-INFO

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

{anyscale-0.24.27 → anyscale-0.24.29}/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
@@ -463,6 +468,11 @@ class AnyscaleClient(AnyscaleClientInterface):
 
         return self.get_default_compute_config(cloud_id=self.get_cloud_id()).id
 
+    def archive_compute_config(self, *, compute_config_id):
+        self._internal_api_client.archive_compute_template_api_v2_compute_templates_compute_template_id_archive_post(
+            compute_config_id
+        )
+
     def get_default_compute_config(self, *, cloud_id: str) -> ClusterCompute:
         return self._external_api_client.get_default_cluster_compute(
             cloud_id=cloud_id,

{anyscale-0.24.27 → anyscale-0.24.29}/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.
 
@@ -144,6 +144,11 @@ class AnyscaleClientInterface(ABC):
         """
         raise NotImplementedError
 
+    @abstractmethod
+    def archive_compute_config(self, *, compute_config_id: str):
+        """Archive the compute config for the provided ID."""
+        raise NotImplementedError
+
     @abstractmethod
     def get_default_compute_config(self, *, cloud_id: str) -> ClusterCompute:
         """Get the default compute config for the provided cloud ID."""

{anyscale-0.24.27 → anyscale-0.24.29}/anyscale/_private/anyscale_client/fake_anyscale_client.py

@@ -68,6 +68,7 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         )
         self._compute_config_id_to_cloud_id: Dict[str, str] = {}
         self._compute_configs: Dict[str, ClusterCompute] = {}
+        self._archived_compute_configs: Dict[str, ClusterCompute] = {}
         self._workspace_cluster: Optional[Cluster] = None
         self._workspace_dependency_tracking_enabled: bool = False
         self._services: Dict[str, ServiceModel] = {}
@@ -239,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)
@@ -252,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:
@@ -260,6 +274,14 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
 
         return self.get_default_compute_config(cloud_id=self.get_cloud_id()).id
 
+    def archive_compute_config(self, *, compute_config_id: str):
+        self._archived_compute_configs[compute_config_id] = self._compute_configs.pop(
+            compute_config_id
+        )
+
+    def is_archived_compute_config(self, compute_config_id: str) -> bool:
+        return compute_config_id in self._archived_compute_configs
+
     def set_default_compute_config(
         self, compute_config: ClusterCompute, *, cloud_id: str
     ):

{anyscale-0.24.27 → anyscale-0.24.29}/anyscale/_private/docgen/__main__.py

@@ -54,6 +54,7 @@ ALL_MODULES = [
         cli_commands=[
             service_commands.deploy,
             service_commands.status,
+            service_commands.wait,
             service_commands.rollback,
             service_commands.terminate,
         ],
@@ -61,6 +62,7 @@ ALL_MODULES = [
         sdk_commands=[
             anyscale.service.deploy,
             anyscale.service.status,
+            anyscale.service.wait,
             anyscale.service.rollback,
             anyscale.service.terminate,
         ],
@@ -79,7 +81,11 @@ ALL_MODULES = [
         cli_prefix="anyscale compute-config",
         cli_commands=[],
         sdk_prefix="anyscale.compute_config",
-        sdk_commands=[anyscale.compute_config.create, anyscale.compute_config.get],
+        sdk_commands=[
+            anyscale.compute_config.create,
+            anyscale.compute_config.get,
+            anyscale.compute_config.archive,
+        ],
         models=[ComputeConfig, HeadNodeConfig, WorkerNodeGroupConfig, MarketType],
     ),
 ]

{anyscale-0.24.27 → anyscale-0.24.29}/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
@@ -356,7 +357,7 @@ class MarkdownGenerator:
                     continue
 
                 assert param.annotation is not inspect.Parameter.empty
-                type_str = self._model_type_to_string(param.annotation)
+                type_str = "(" + self._model_type_to_string(param.annotation) + ")"
                 if param.default != inspect.Parameter.empty:
                     type_str += f" = {param.default!s}"
 
@@ -366,7 +367,7 @@ class MarkdownGenerator:
                         f"SDK command '{sdk_prefix}.{c.__name__}' is missing a docstring for argument '{name}'"
                     )
 
-                md += f"- **`{name}` ({type_str})**: {arg_docs}"
+                md += f"- **`{name}` {type_str}**: {arg_docs}"
                 md += "\n"
             md += "\n"
 

{anyscale-0.24.27 → anyscale-0.24.29}/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.27 → anyscale-0.24.29}/anyscale/_private/sdk/base_sdk.py

@@ -4,6 +4,7 @@ from anyscale._private.anyscale_client import (
     AnyscaleClient,
     AnyscaleClientInterface,
 )
+from anyscale._private.sdk.timer import RealTimer, Timer
 from anyscale.cli_logger import BlockLogger
 
 
@@ -15,9 +16,11 @@ class BaseSDK:
         *,
         logger: Optional[BlockLogger] = None,
         client: Optional[AnyscaleClientInterface] = None,
+        timer: Optional[Timer] = None,
     ):
         self._logger = logger or BlockLogger()
         self._client = client or AnyscaleClient()
+        self._timer = timer or RealTimer()
 
     @property
     def logger(self) -> BlockLogger:
@@ -26,3 +29,7 @@ class BaseSDK:
     @property
     def client(self) -> AnyscaleClientInterface:
         return self._client
+
+    @property
+    def timer(self) -> Timer:
+        return self._timer

anyscale-0.24.29/anyscale/_private/sdk/timer.py

@@ -0,0 +1,46 @@
+from abc import ABC, abstractmethod
+import time
+from typing import Callable, Iterator, Optional
+
+
+class Timer(ABC):
+    @abstractmethod
+    def poll(self, *, timeout_s: float, interval_s: float) -> Iterator[None]:
+        """Yield every interval_s until the timeout.
+
+        Exits when timeout_s is reached. DOES NOT raise TimeoutError.
+        """
+
+
+class RealTimer(Timer):
+    def poll(self, *, timeout_s: float, interval_s: float) -> Iterator[None]:
+        start_time_s = time.time()
+        while True:
+            yield
+
+            if time.time() - start_time_s >= timeout_s:
+                break
+            else:
+                time.sleep(interval_s)
+
+
+class FakeTimer(Timer):
+    def __init__(self):
+        self._on_poll_iteration: Optional[Callable[[int], None]] = None
+
+    def set_on_poll_iteration(self, on_poll_iteration: Callable[[int], None]):
+        self._on_poll_iteration = on_poll_iteration
+
+    def poll(self, *, timeout_s: float, interval_s: float) -> Iterator[None]:
+        iteration: int = 0
+        elapsed_time_s: float = 0
+        while True:
+            if self._on_poll_iteration is not None:
+                self._on_poll_iteration(iteration)
+            iteration += 1
+
+            yield
+
+            elapsed_time_s += interval_s
+            if elapsed_time_s >= timeout_s:
+                break

{anyscale-0.24.27 → anyscale-0.24.29}/anyscale/_private/workload/workload_sdk.py

@@ -7,7 +7,7 @@ from anyscale._private.anyscale_client import (
     AnyscaleClientInterface,
     WORKSPACE_CLUSTER_NAME_PREFIX,
 )
-from anyscale._private.sdk.base_sdk import BaseSDK
+from anyscale._private.sdk.base_sdk import BaseSDK, Timer
 from anyscale.cli_logger import BlockLogger
 from anyscale.compute_config._private.compute_config_sdk import ComputeConfigSDK
 from anyscale.compute_config.models import ComputeConfig
@@ -27,8 +27,9 @@ class WorkloadSDK(BaseSDK):
         *,
         logger: Optional[BlockLogger] = None,
         client: Optional[AnyscaleClientInterface] = None,
+        timer: Optional[Timer] = None,
     ):
-        super().__init__(logger=logger, client=client)
+        super().__init__(logger=logger, client=client, timer=timer)
         self._enable_image_build_for_tracked_requirements = (
             ENABLE_IMAGE_BUILD_FOR_TRACKED_REQUIREMENTS
         )

{anyscale-0.24.27 → anyscale-0.24.29}/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.27 → anyscale-0.24.29}/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.27 → anyscale-0.24.29}/anyscale/commands/machine_commands.py

@@ -18,8 +18,6 @@ def machine_cli() -> None:
 
 
 @machine_cli.command(name="list", help="List machines registered to Anyscale.")
-@click.option("--cloud", type=str, help="Provide a cloud name.")
-@click.option("--cloud-id", type=str, help="Provide a cloud ID.")
 @click.option("--machine-pool-name", type=str, help="Provide a machine pool name.")
 @click.option(
     "--format",
@@ -28,22 +26,9 @@ def machine_cli() -> None:
     default="table",
     help="The format to return results in.",
 )
-def list_machines(
-    cloud: Optional[str],
-    cloud_id: Optional[str],
-    machine_pool_name: Optional[str],
-    format_: str,
-) -> None:
-
-    if not cloud and not cloud_id and not machine_pool_name:
-        raise ValueError(
-            "One of {'cloud', 'cloud-id', 'machine-pool-name'} must be provided."
-        )
-
+def list_machines(machine_pool_name: str, format_: str,) -> None:
     machines_controller = MachineController()
-    output = machines_controller.list_machines(
-        cloud=cloud, cloud_id=cloud_id, machine_pool_name=machine_pool_name
-    )
+    output = machines_controller.list_machines(machine_pool_name)
 
     if format_ == "json":
         rows = []

{anyscale-0.24.27 → anyscale-0.24.29}/anyscale/commands/service_commands.py

@@ -11,7 +11,7 @@ from anyscale.cli_logger import BlockLogger
 from anyscale.commands.util import convert_kv_strings_to_dict
 from anyscale.controllers.service_controller import ServiceController
 import anyscale.service
-from anyscale.service import ServiceConfig, ServiceStatus
+from anyscale.service import ServiceConfig, ServiceState, ServiceStatus
 from anyscale.util import validate_non_negative_arg
 
 
@@ -312,6 +312,55 @@ def status(name: Optional[str], config_file: Optional[str], json: bool, verbose:
         print(stream.getvalue(), end="")
 
 
+@service_cli.command(
+    name="wait", help="Wait for a service to enter a target state.", hidden=True,
+)
+@click.option(
+    "-n", "--name", required=False, default=None, type=str, help="Name of the service.",
+)
+@click.option(
+    "-f",
+    "--config-file",
+    required=False,
+    default=None,
+    type=str,
+    help="Path to a YAML config file to read the name from.",
+)
+@click.option(
+    "-s",
+    "--state",
+    default=str(ServiceState.RUNNING),
+    help="The ServiceState to wait for the service to reach. Defaults to RUNNING.",
+)
+@click.option(
+    "-t",
+    "--timeout-s",
+    default=600,
+    type=float,
+    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."
+        )
+
+    if config_file is not None:
+        name = _read_name_from_config_file(config_file)
+
+    if name is None:
+        raise click.ClickException(
+            "Service name must be provided using '--name' or in a config file using '-f'."
+        )
+
+    anyscale.service.wait(name=name, state=state, timeout_s=timeout_s)
+
+
 @service_cli.command(
     name="rollout", help="Roll out a service.",
 )
@@ -463,10 +512,7 @@ def rollback(
     config_file: Optional[str],
     max_surge_percent: Optional[int],
 ):
-    """Perform a rollback for a service that is currently in a rollout.
-
-    This *only* applies to v2 services.
-    """
+    """Perform a rollback for a service that is currently in a rollout."""
     service_controller = ServiceController()
     service_id = service_controller.get_service_id(
         service_id=service_id,