Add pod name variable

G-Research · Mar 26, 2024 · eb0f33e · eb0f33e
1 parent 96bcf63
commit eb0f33e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...main/scala/org/apache/spark/scheduler/cluster/k8s/KubernetesClusterSchedulerBackend.scala b/...main/scala/org/apache/spark/scheduler/cluster/k8s/KubernetesClusterSchedulerBackend.scala
@@ -105,7 +105,8 @@ private[spark] class KubernetesClusterSchedulerBackend(
   override def getDriverAttributes: Option[Map[String, String]] = Some(Map(
     "LOG_FILES" -> "log",
     "APP_ID" -> applicationId,
-    "KUBERNETES_NAMESPACE" -> namespace
+    "KUBERNETES_NAMESPACE" -> namespace,
+    "KUBERNETES_POD_NAME" -> System.getenv(ENV_DRIVER_POD_NAME)
   ))
 
   override def start(): Unit = {