diff --git a/airflow-core/src/airflow/jobs/scheduler_job_runner.py b/airflow-core/src/airflow/jobs/scheduler_job_runner.py index b5535ea44a6a5..514906e76c49f 100644 --- a/airflow-core/src/airflow/jobs/scheduler_job_runner.py +++ b/airflow-core/src/airflow/jobs/scheduler_job_runner.py @@ -2188,6 +2188,9 @@ def _find_task_instances_without_heartbeats(self, *, session: Session) -> list[T limit_dttm = timezone.utcnow() - timedelta(seconds=self._task_instance_heartbeat_timeout_secs) task_instances_without_heartbeats = session.scalars( select(TI) + .options(selectinload(TI.dag_model)) + .options(selectinload(TI.dag_run)) + .options(selectinload(TI.dag_version)) .with_hint(TI, "USE INDEX (ti_state)", dialect_name="mysql") .join(DM, TI.dag_id == DM.dag_id) .where(