Changes

dask · Sep 23, 2023 · 98c02fa · 98c02fa
1 parent 2f98f6d
commit 98c02fa
Showing 1 changed file with 6 additions and 9 deletions.
diff --git a/dask_kubernetes/operator/controller/controller.py b/dask_kubernetes/operator/controller/controller.py
@@ -72,7 +72,7 @@ def build_scheduler_deployment_spec(
         **{
             "dask.org/cluster-name": cluster_name,
             "dask.org/component": "scheduler",
-            "dask.org/object-name": name,
+            "dask.org/component-name": name,
             "sidecar.istio.io/inject": "false",
         }
     )
@@ -104,7 +104,7 @@ def build_scheduler_service_spec(cluster_name, spec, annotations, labels):
         **{
             "dask.org/cluster-name": cluster_name,
             "dask.org/component": "scheduler",
-            "dask.org/object-name": name,
+            "dask.org/component-name": name,
         }
     )
     return {
@@ -128,7 +128,7 @@ def build_worker_deployment_spec(
             "dask.org/cluster-name": cluster_name,
             "dask.org/workergroup-name": worker_group_name,
             "dask.org/component": "worker",
-            "dask.org/object-name": worker_name,
+            "dask.org/component-name": worker_name,
             "sidecar.istio.io/inject": "false",
         }
     )
@@ -183,7 +183,7 @@ def build_job_pod_spec(job_name, cluster_name, namespace, spec, annotations, lab
         **{
             "dask.org/cluster-name": cluster_name,
             "dask.org/component": "job-runner",
-            "dask.org/object-name": name,
+            "dask.org/component-name": name,
             "sidecar.istio.io/inject": "false",
         }
     )
@@ -217,7 +217,7 @@ def build_default_worker_group_spec(cluster_name, spec, annotations, labels):
         **{
             "dask.org/cluster-name": cluster_name,
             "dask.org/component": "workergroup",
-            "dask.org/object-name": name,
+            "dask.org/component-name": name,
         }
     )
     return {
@@ -239,7 +239,7 @@ def build_cluster_spec(name, worker_spec, scheduler_spec, annotations, labels):
     labels.update(
         **{
             "dask.org/cluster-name": name,
-            "dask.org/object-name": name,
+            "dask.org/component-name": name,
         }
     )
     return {
@@ -624,9 +624,6 @@ async def daskworkergroup_replica_update(
                 if (
                     worker_pod.status.phase
                     not in [
-                        "ContainerCreating",
-                        "Initializing",
-                        "Terminating",
                         "Running",
                     ]
                     and not await worker.ready()