diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 81e24f2e8a3b3..bb599348827fc 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -413,6 +413,7 @@ repos: ^airflow-ctl.*\.py$| ^airflow-core/src/airflow/models/.*\.py$| ^airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_assets.py$| + ^airflow-core/tests/unit/ti_deps/deps/test_runnable_exec_date_dep.py$| ^task_sdk.*\.py$ pass_filenames: true - id: update-supported-versions diff --git a/airflow-core/tests/unit/ti_deps/deps/test_runnable_exec_date_dep.py b/airflow-core/tests/unit/ti_deps/deps/test_runnable_exec_date_dep.py index bd0a2b6836a8d..771d55b57a8d5 100644 --- a/airflow-core/tests/unit/ti_deps/deps/test_runnable_exec_date_dep.py +++ b/airflow-core/tests/unit/ti_deps/deps/test_runnable_exec_date_dep.py @@ -21,6 +21,7 @@ import pytest import time_machine +from sqlalchemy import delete from airflow._shared.timezones.timezone import datetime from airflow.models import DagRun, TaskInstance @@ -33,8 +34,8 @@ @pytest.fixture(autouse=True) def clean_db(session): yield - session.query(DagRun).delete() - session.query(TaskInstance).delete() + session.execute(delete(DagRun)) + session.execute(delete(TaskInstance)) @time_machine.travel("2016-11-01")