fix: sqa deprecations for airflow task cmd

apache · Apr 24, 2024 · 3052622 · 3052622
1 parent 9619536
commit 3052622
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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 create_if_necessary == "db":
+            session.add(ti)
         ti.dag_run = dag_run
     else:
         ti = ti_or_none