From cb03e440af06837519a9cf9c778e98984819da08 Mon Sep 17 00:00:00 2001
From: Sebastian Daum <daumse@mediamarktsaturn.com>
Date: Wed, 24 Apr 2024 23:07:12 +0200
Subject: [PATCH] fix: sqa deprecations for airflow task cmd

---
 airflow/cli/commands/task_command.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/airflow/cli/commands/task_command.py b/airflow/cli/commands/task_command.py
index 639ab74b8df8a..7cde8f478e3be 100644
--- a/airflow/cli/commands/task_command.py
+++ b/airflow/cli/commands/task_command.py
@@ -199,6 +199,8 @@ def _get_ti_db_access(
             )
         # TODO: Validate map_index is in range?
         ti = TaskInstance(task, run_id=dag_run.run_id, map_index=map_index)
+        if dag_run in session:
+            session.add(ti)
         ti.dag_run = dag_run
     else:
         ti = ti_or_none