diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 46d505f55ab68..792168d05356b 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/api_fastapi/core_api/routes/public/test_hitl.py$| ^providers/openlineage/.*\.py$| ^task_sdk.*\.py$ pass_filenames: true diff --git a/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_hitl.py b/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_hitl.py index d9b5a0287bb4b..93d8c4f63d472 100644 --- a/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_hitl.py +++ b/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_hitl.py @@ -25,7 +25,7 @@ import pytest import time_machine -from sqlalchemy import select +from sqlalchemy import delete, select from sqlalchemy.orm import Session from airflow._shared.timezones.timezone import utc, utcnow @@ -275,7 +275,7 @@ def expected_sample_hitl_detail_dict(sample_ti: TaskInstance) -> dict[str, Any]: @pytest.fixture(autouse=True) def cleanup_audit_log(session: Session) -> None: - session.query(Log).delete() + session.execute(delete(Log)) session.commit()