From aafcc372af84f8776a85bfa8f1eeede2ca68e32e Mon Sep 17 00:00:00 2001 From: Jens Scheffler Date: Sat, 11 Jan 2025 19:26:48 +0100 Subject: [PATCH] Remove type hints only working in Airflow 3 --- tests/api/common/test_mark_tasks.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/api/common/test_mark_tasks.py b/tests/api/common/test_mark_tasks.py index 124232842d9a0..0cf58ee74a67a 100644 --- 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")