Fix import

apache · Nov 14, 2024 · 14a4cff · 14a4cff
1 parent 3f945fb
commit 14a4cff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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