This is an automated email from the ASF dual-hosted git repository.
potiuk pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new 7bb1d4a117 Use class-bound attribute directly in SA `lazyload` (#39195)
7bb1d4a117 is described below
commit 7bb1d4a11752423e63609d237092aa1152b4912b
Author: Andrey Anshin <andrey.ans...@taragol.is>
AuthorDate: Tue Apr 23 14:01:42 2024 +0400
Use class-bound attribute directly in SA `lazyload` (#39195)
---
airflow/jobs/scheduler_job_runner.py | 2 +-
airflow/models/taskinstance.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/airflow/jobs/scheduler_job_runner.py
b/airflow/jobs/scheduler_job_runner.py
index 2d2c657dac..631de5692e 100644
--- a/airflow/jobs/scheduler_job_runner.py
+++ b/airflow/jobs/scheduler_job_runner.py
@@ -1631,7 +1631,7 @@ class SchedulerJobRunner(BaseJobRunner, LoggingMixin):
query = (
select(TI)
- .options(lazyload("dag_run")) # avoids double join to
dag_run
+ .options(lazyload(TI.dag_run)) # avoids double join
to dag_run
.where(TI.state.in_(State.adoptable_states))
.join(TI.queued_by_job)
.where(Job.state.is_distinct_from(JobState.RUNNING))
diff --git a/airflow/models/taskinstance.py b/airflow/models/taskinstance.py
index b4d5d5d65a..cb071e2edc 100644
--- a/airflow/models/taskinstance.py
+++ b/airflow/models/taskinstance.py
@@ -1854,7 +1854,7 @@ class TaskInstance(Base, LoggingMixin):
) -> TaskInstance | TaskInstancePydantic | None:
query = (
session.query(TaskInstance)
- .options(lazyload("dag_run")) # lazy load dag run to avoid
locking it
+ .options(lazyload(TaskInstance.dag_run)) # lazy load dag run to
avoid locking it
.filter_by(
dag_id=dag_id,
run_id=run_id,
