Ensure airflow.macros is being loaded properly in subprocesses

RikHeijdens · Dec 4, 2020 · b2c035c · b2c035c
1 parent b2258a4
commit b2c035c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/airflow/utils/python_virtualenv_script.jinja2 b/airflow/utils/python_virtualenv_script.jinja2
@@ -20,6 +20,10 @@
 import {{ pickling_library }}
 import sys
 
+# Ensure airflow.macros has been loaded properly.
+import airflow.plugins_manager
+airflow.plugins_manager.integrate_macros_plugins()
+
 # Read args
 {% if op_args or op_kwargs %}
 with open(sys.argv[1], "rb") as file: