Remove type hints only working in Airflow 3

apache · Jan 11, 2025 · aafcc37 · aafcc37
1 parent fa03210
commit aafcc37
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/api/common/test_mark_tasks.py b/tests/api/common/test_mark_tasks.py
@@ -27,12 +27,10 @@
 if TYPE_CHECKING:
     from airflow.models.taskinstance import TaskInstance
 
-    from tests_common.pytest_plugin import DagMaker
-
 pytestmark = pytest.mark.db_test
 
 
-def test_set_dag_run_state_to_failed(dag_maker: DagMaker):
+def test_set_dag_run_state_to_failed(dag_maker):
     with dag_maker("TEST_DAG_1"):
         with EmptyOperator(task_id="teardown").as_teardown():
             EmptyOperator(task_id="running")
@@ -54,7 +52,7 @@ def test_set_dag_run_state_to_failed(dag_maker: DagMaker):
     assert "teardown" not in task_dict
 
 
-def test_set_dag_run_state_to_success(dag_maker: DagMaker):
+def test_set_dag_run_state_to_success(dag_maker):
     with dag_maker("TEST_DAG_1"):
         with EmptyOperator(task_id="teardown").as_teardown():
             EmptyOperator(task_id="running")