diff --git a/dags/migrations/migrations.py b/dags/migrations/migrations.py index 86f3161..6d01466 100644 --- 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, )