From 412e1cf9f11b4d976523e5875e598fdaec4ed84f Mon Sep 17 00:00:00 2001 From: Jacob Tomlinson Date: Tue, 7 Jun 2022 16:35:23 +0100 Subject: [PATCH] Fix dashboard forwarding --- dask_kubernetes/classic/kubecluster.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/dask_kubernetes/classic/kubecluster.py b/dask_kubernetes/classic/kubecluster.py index 12ae89c82..08335e16e 100644 --- a/dask_kubernetes/classic/kubecluster.py +++ b/dask_kubernetes/classic/kubecluster.py @@ -31,7 +31,7 @@ ) from ..common.networking import ( get_external_address_for_scheduler_service, - port_forward_dashboard, + get_scheduler_address, ) logger = logging.getLogger(__name__) @@ -644,9 +644,12 @@ async def _start(self): await super()._start() - self.forwarded_dashboard_port = await port_forward_dashboard( - self.name, self.namespace + dashboard_address = await get_scheduler_address( + self.scheduler.service.metadata.name, + self.namespace, + port_name="http-dashboard", ) + self.forwarded_dashboard_port = dashboard_address.split(":")[-1] @classmethod def from_dict(cls, pod_spec, **kwargs):