Add KubeCluster.jupyter_link property

dask · Aug 14, 2023 · 31785b7 · 31785b7
1 parent d0c6e9a
commit 31785b7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/dask_kubernetes/cli/cli.py b/dask_kubernetes/cli/cli.py
@@ -97,9 +97,7 @@ def port_forward(cluster):
         console.print(f"Scheduler at: [magenta][not bold]{kcluster.scheduler_address}")
         console.print(f"Dashboard at: [cyan][not bold]{kcluster.dashboard_link}")
         if kcluster.jupyter:
-            console.print(
-                f"Jupyter at: [orange3][not bold]{kcluster.dashboard_link.replace('/status', '/jupyter/lab')}"
-            )
+            console.print(f"Jupyter at: [orange3][not bold]{kcluster.jupyter_link}")
         console.print("Press ctrl+c to exit", style="bright_black")
         while True:
             time.sleep(0.1)

diff --git a/dask_kubernetes/operator/kubecluster/kubecluster.py b/dask_kubernetes/operator/kubecluster/kubecluster.py
@@ -818,6 +818,12 @@ def from_name(cls, name, **kwargs):
             **kwargs,
         )
 
+    @property
+    def jupyter_link(self):
+        if self.jupyter:
+            return self.dashboard_link.replace("/status", "/jupyter/lab")
+        raise RuntimeError("KubeCluster not started with jupyter enabled")
+
 
 def make_cluster_spec(
     name,