From 46350db5b98c07d6018c9156fecae74b14f19db1 Mon Sep 17 00:00:00 2001 From: Ross MacDonald <42966821+RossmacD@users.noreply.github.com> Date: Fri, 25 Oct 2024 14:21:42 +0100 Subject: [PATCH] Improve retire_worker fallback by using deployments instead of pods (#912) --- dask_kubernetes/operator/controller/controller.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dask_kubernetes/operator/controller/controller.py b/dask_kubernetes/operator/controller/controller.py index ce1f5a0a..19599e59 100644 --- a/dask_kubernetes/operator/controller/controller.py +++ b/dask_kubernetes/operator/controller/controller.py @@ -488,7 +488,7 @@ async def retire_workers( "This can result in lost data, see https://kubernetes.dask.org/en/latest/operator_troubleshooting.html." ) workers = await kr8s.asyncio.get( - "pods", + "deployments", namespace=namespace, label_selector={"dask.org/workergroup-name": worker_group_name}, )