OA: added kubernetes executor in migration DAG

cern-sis · Apr 18, 2024 · 8bd2528 · 8bd2528
1 parent 25a121a
commit 8bd2528
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dags/migrations/migrations.py b/dags/migrations/migrations.py
@@ -4,6 +4,7 @@
 from airflow.models import DAG
 from airflow.models.param import Param
 from airflow_provider_alembic.operators.alembic import AlembicOperator
+from executor_config import kubernetes_executor_config
 
 with DAG(
     "migrations",
@@ -17,4 +18,5 @@
         command="{{ params.command }}",
         revision="{{ params.revision }}",
         script_location=f"{os.environ['AIRFLOW_HOME']}/dags/migrations/",
+        executor_config=kubernetes_executor_config,
     )