From 851b89720fe244060762cf5d87783d2828ec13af Mon Sep 17 00:00:00 2001
From: Jens Scheffler <jens.scheffler@de.bosch.com>
Date: Thu, 19 Dec 2024 11:08:10 +0100
Subject: [PATCH] Add missing session to CLI call
---
airflow/cli/commands/remote_commands/task_command.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/airflow/cli/commands/remote_commands/task_command.py b/airflow/cli/commands/remote_commands/task_command.py
index e4c72b94d628d8..ad3c26ff56eda3 100644
--- a/airflow/cli/commands/remote_commands/task_command.py
+++ b/airflow/cli/commands/remote_commands/task_command.py
@@ -286,7 +286,8 @@ def _run_task_by_executor(args, dag: DAG, ti: TaskInstance) -> None:
from airflow.executors import workloads
workload = workloads.ExecuteTask.make(ti, dag_path=dag.relative_fileloc)
- executor.queue_workload(workload)
+ with create_session() as session:
+ executor.queue_workload(workload, session)
else:
executor.queue_task_instance(
ti,