Skip to content

Commit

Permalink
Add rich status for endpoint fetching
Browse files Browse the repository at this point in the history
  • Loading branch information
Michaelvll committed Jul 4, 2024
1 parent b12a8c3 commit 422e9f4
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion sky/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@
from sky.skylet import job_lib
from sky.usage import usage_lib
from sky.utils import controller_utils
from sky.utils import rich_utils
from sky.utils import subprocess_utils

if typing.TYPE_CHECKING:
Expand Down Expand Up @@ -126,7 +127,8 @@ def endpoints(cluster: str,
RuntimeError: if the cluster has no ports to be exposed or no endpoints
are exposed yet.
"""
return backend_utils.get_endpoints(cluster=cluster, port=port)
with rich_utils.safe_status('[bold cyan]Fetching endpoints...[/]'):
return backend_utils.get_endpoints(cluster=cluster, port=port)


@usage_lib.entrypoint
Expand Down

0 comments on commit 422e9f4

Please sign in to comment.