Allow passing empty labels in the driver config

apache · Jan 23, 2025 · bbdcb82 · bbdcb82
1 parent c092d44
commit bbdcb82
Showing 1 changed file with 1 addition and 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
@@ -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,
                 )