From bbdcb820332f39affbc98257f7942cab9a958abb Mon Sep 17 00:00:00 2001 From: Maxim Logvinenko Date: Tue, 21 Jan 2025 20:42:25 +0000 Subject: [PATCH] Allow passing empty labels in the driver config --- .../cncf/kubernetes/operators/custom_object_launcher.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/providers/src/airflow/providers/cncf/kubernetes/operators/custom_object_launcher.py b/providers/src/airflow/providers/cncf/kubernetes/operators/custom_object_launcher.py index 8e2edc260670b9..c6d1d14cc624e8 100644 --- a/providers/src/airflow/providers/cncf/kubernetes/operators/custom_object_launcher.py +++ b/providers/src/airflow/providers/cncf/kubernetes/operators/custom_object_launcher.py @@ -291,7 +291,7 @@ def start_spark_job(self, image=None, code_path=None, startup_timeout: int = 600 # Wait for the driver pod to come alive self.pod_spec = k8s.V1Pod( metadata=k8s.V1ObjectMeta( - labels=self.spark_obj_spec["spec"]["driver"]["labels"], + labels=self.spark_obj_spec["spec"]["driver"].get("labels"), name=self.spark_obj_spec["metadata"]["name"] + "-driver", namespace=self.namespace, )