Fix EdgeExecutor breeze call (#43842)

apache · Nov 9, 2024 · 229750d · 229750d
1 parent 9bbf6f1
commit 229750d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/dev/breeze/src/airflow_breeze/params/shell_params.py b/dev/breeze/src/airflow_breeze/params/shell_params.py
@@ -497,6 +497,9 @@ def env_variables_for_docker_commands(self) -> dict[str, str]:
         _set_var(_env, "AIRFLOW__CELERY__BROKER_URL", self.airflow_celery_broker_url)
         _set_var(_env, "AIRFLOW__CORE__EXECUTOR", self.executor)
         if self.executor == EDGE_EXECUTOR:
+            _set_var(
+                _env, "AIRFLOW__CORE__EXECUTOR", "airflow.providers.edge.executors.edge_executor.EdgeExecutor"
+            )
             _set_var(_env, "AIRFLOW__EDGE__API_ENABLED", "true")
             _set_var(_env, "AIRFLOW__EDGE__API_URL", "http://localhost:8080/edge_worker/v1/rpcapi")
         _set_var(_env, "ANSWER", get_forced_answer() or "")