diff --git a/airflow/ti_deps/deps/pool_slots_available_dep.py b/airflow/ti_deps/deps/pool_slots_available_dep.py index f8c461e51f0d1..a37e62aac0471 100644 --- a/airflow/ti_deps/deps/pool_slots_available_dep.py +++ b/airflow/ti_deps/deps/pool_slots_available_dep.py @@ -55,7 +55,7 @@ def _get_dep_statuses(self, ti, session, dep_context=None): else: # Controlled by UNIQUE key in slot_pool table, # only one result can be returned. - open_slots = pools[0].open_slots() + open_slots = pools[0].open_slots(session=session) if ti.state in EXECUTION_STATES: open_slots += ti.pool_slots diff --git a/tests/jobs/test_local_task_job.py b/tests/jobs/test_local_task_job.py index c99f0435b88b1..8a4df12db305f 100644 --- a/tests/jobs/test_local_task_job.py +++ b/tests/jobs/test_local_task_job.py @@ -879,5 +879,5 @@ def test_number_of_queries_single_loop(mock_get_task_runner, return_codes, dag_m ti.refresh_from_task(task) job = LocalTaskJob(task_instance=ti, executor=MockExecutor()) - with assert_queries_count(20): + with assert_queries_count(18): job.run()