From 07a23cfc709a30951edebd4adda459cee43d8b93 Mon Sep 17 00:00:00 2001
From: Sebastian Daum <daumse@mediamarktsaturn.com>
Date: Sat, 4 May 2024 20:39:44 +0200
Subject: [PATCH] Resolve `RemovedIn20Warning` in `airflow task` command
 (#39244)

---
 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 639ab74b8df8a3..7cde8f478e3beb 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