Fix fetching single running pod (#846) (#847)

Co-authored-by: Jacob Tomlinson <[email protected]>
dask · Jan 5, 2024 · aae2688 · aae2688
1 parent 90c7d9b
commit aae2688
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/dask_kubernetes/common/networking.py b/dask_kubernetes/common/networking.py
@@ -204,6 +204,7 @@ async def wait_for_scheduler(cluster_name, namespace, timeout=None):
         try:
             pod = await Pod.get(
                 label_selector=f"dask.org/component=scheduler,dask.org/cluster-name={cluster_name}",
+                field_selector="status.phase=Running",
                 namespace=namespace,
             )
         except kr8s.NotFoundError:

diff --git a/dask_kubernetes/operator/networking.py b/dask_kubernetes/operator/networking.py
@@ -203,6 +203,7 @@ async def wait_for_scheduler(cluster_name, namespace, timeout=None):
         try:
             pod = await Pod.get(
                 label_selector=f"dask.org/component=scheduler,dask.org/cluster-name={cluster_name}",
+                field_selector="status.phase=Running",
                 namespace=namespace,
             )
         except kr8s.NotFoundError: