From 6a272167d9d017e3f00aa8b4f558344f2992ee01 Mon Sep 17 00:00:00 2001 From: Derek Yu Date: Wed, 24 Jan 2024 18:30:20 -0800 Subject: [PATCH] add back cluster auth --- dask_kubernetes/operator/controller/controller.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/dask_kubernetes/operator/controller/controller.py b/dask_kubernetes/operator/controller/controller.py index 00c47aeed..f298e9102 100644 --- a/dask_kubernetes/operator/controller/controller.py +++ b/dask_kubernetes/operator/controller/controller.py @@ -20,6 +20,7 @@ DaskJob, DaskWorkerGroup, ) +from dask_kubernetes.common.auth import ClusterAuth from dask_kubernetes.operator.networking import get_scheduler_address _ANNOTATION_NAMESPACES_TO_IGNORE = ( @@ -245,6 +246,9 @@ def build_cluster_spec(name, worker_spec, scheduler_spec, annotations, labels): @kopf.on.startup() async def startup(settings: kopf.OperatorSettings, **kwargs): + # Authenticate with k8s + await ClusterAuth.load_first() + # Set server and client timeouts to reconnect from time to time. # In rare occasions the connection might go idle we will no longer receive any events. # These timeouts should help in those cases.