From 14a4cffb10f1055e79e0236ccff91aa8e28a1b6a Mon Sep 17 00:00:00 2001 From: Tzu-ping Chung Date: Thu, 14 Nov 2024 15:19:49 +0800 Subject: [PATCH] Fix import --- tests/sensors/test_external_task_sensor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/sensors/test_external_task_sensor.py b/tests/sensors/test_external_task_sensor.py index 67970ba52ade..4cd7b5e5f8c6 100644 --- a/tests/sensors/test_external_task_sensor.py +++ b/tests/sensors/test_external_task_sensor.py @@ -28,7 +28,7 @@ import pytest -from airflow import __version__ as exceptions, settings +from airflow import settings from airflow.decorators import task as task_deco from airflow.exceptions import AirflowException, AirflowSensorTimeout, AirflowSkipException, TaskDeferred from airflow.models import DagBag, DagRun, TaskInstance @@ -661,7 +661,7 @@ def test_external_task_sensor_fn(self): poke_interval=1, dag=self.dag, ) - with pytest.raises(exceptions.AirflowSensorTimeout): + with pytest.raises(AirflowSensorTimeout): op2.run(start_date=DEFAULT_DATE, end_date=DEFAULT_DATE, ignore_ti_state=True) @pytest.mark.skip_if_database_isolation_mode # Test is broken in db isolation mode