no alias

apache · Mar 19, 2024 · fbe138d · fbe138d
1 parent efc2582
commit fbe138d
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/airflow/models/taskinstance.py b/airflow/models/taskinstance.py
@@ -1842,15 +1842,14 @@ def key(self) -> TaskInstanceKey:
     @staticmethod
     @internal_api_call
     def _set_state(ti: TaskInstance | TaskInstancePydantic, state, session: Session) -> bool:
-        TI = TaskInstance
         if not isinstance(ti, TaskInstance):
             ti = (
-                session.query(TI)
+                session.query(TaskInstance)
                 .filter(
-                    TI.task_id == ti.task_id,
-                    TI.dag_id == ti.dag_id,
-                    TI.run_id == ti.run_id,
-                    TI.map_index == ti.map_index,
+                    TaskInstance.task_id == ti.task_id,
+                    TaskInstance.dag_id == ti.dag_id,
+                    TaskInstance.run_id == ti.run_id,
+                    TaskInstance.map_index == ti.map_index,
                 )
                 .one()
             )