Export scheduler env variable into worker pods. (#38)

twitter-forks · Apr 9, 2020 · 8a689af · 8a689af
1 parent f7050fb
commit 8a689af
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/airflow/contrib/kubernetes/worker_configuration.py b/airflow/contrib/kubernetes/worker_configuration.py
@@ -153,6 +153,13 @@ def _get_environment(self):
                 self.kube_config.git_subpath     # dags
             )
             env['AIRFLOW__CORE__DAGS_FOLDER'] = dag_volume_mount_path
+
+        # TODO This change can be submitted into the apache as well.
+        # Set the scheduler env into the worker pod.
+        os_env = os.environ
+        os_env.update(env)
+        env = os_env
+
         return env
 
     def _get_configmaps(self):