From b04c69a329133822ee2f620d1f590d5a2c990733 Mon Sep 17 00:00:00 2001
From: Zhanghao Wu <zhanghao.wu@outlook.com>
Date: Thu, 4 Jul 2024 06:56:26 +0000
Subject: [PATCH] Add rich status for waiting for the endpoint

---
 sky/core.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/sky/core.py b/sky/core.py
index f043dedbb5e..6b18fd2c190 100644
--- a/sky/core.py
+++ b/sky/core.py
@@ -127,7 +127,8 @@ def endpoints(cluster: str,
         RuntimeError: if the cluster has no ports to be exposed or no endpoints
             are exposed yet.
     """
-    with rich_utils.safe_status('[bold cyan]Fetching endpoints...[/]'):
+    with rich_utils.safe_status('[bold cyan]Fetching endpoints for cluster '
+                                f'{cluster}...[/]'):
         return backend_utils.get_endpoints(cluster=cluster, port=port)