diff --git a/providers/standard/src/airflow/providers/standard/operators/branch.py b/providers/standard/src/airflow/providers/standard/operators/branch.py index 98c059ac38eac..985a1cb59e1cf 100644 --- a/providers/standard/src/airflow/providers/standard/operators/branch.py +++ b/providers/standard/src/airflow/providers/standard/operators/branch.py @@ -27,7 +27,7 @@ if AIRFLOW_V_3_0_PLUS: from airflow.providers.standard.utils.skipmixin import SkipMixin else: - from airflow.models.skipmixin import SkipMixin # type: ignore[no-redef] + from airflow.models.skipmixin import SkipMixin if TYPE_CHECKING: from airflow.sdk.definitions.context import Context diff --git a/providers/standard/src/airflow/providers/standard/operators/python.py b/providers/standard/src/airflow/providers/standard/operators/python.py index 70b3d00785e5c..e0276e4ebe261 100644 --- a/providers/standard/src/airflow/providers/standard/operators/python.py +++ b/providers/standard/src/airflow/providers/standard/operators/python.py @@ -62,7 +62,7 @@ from airflow.providers.standard.operators.branch import BaseBranchOperator from airflow.providers.standard.utils.skipmixin import SkipMixin else: - from airflow.models.skipmixin import SkipMixin # type: ignore[no-redef] + from airflow.models.skipmixin import SkipMixin from airflow.operators.branch import BaseBranchOperator # type: ignore[no-redef] @@ -330,7 +330,7 @@ def get_tasks_to_skip(): self.skip( dag_run=context["dag_run"], tasks=to_skip, - execution_date=cast("DateTime", dag_run.logical_date), # type: ignore[call-arg, union-attr] + execution_date=cast("DateTime", dag_run.logical_date), # type: ignore[call-arg] map_index=context["ti"].map_index, ) diff --git a/providers/standard/src/airflow/providers/standard/operators/trigger_dagrun.py b/providers/standard/src/airflow/providers/standard/operators/trigger_dagrun.py index 7618ef16b2927..3d898b1857ce9 100644 --- a/providers/standard/src/airflow/providers/standard/operators/trigger_dagrun.py +++ b/providers/standard/src/airflow/providers/standard/operators/trigger_dagrun.py @@ -61,7 +61,7 @@ if AIRFLOW_V_3_0_PLUS: from airflow.sdk.execution_time.xcom import XCom else: - from airflow.models import XCom # type: ignore[no-redef] + from airflow.models import XCom class DagIsPaused(AirflowException): diff --git a/providers/standard/src/airflow/providers/standard/sensors/external_task.py b/providers/standard/src/airflow/providers/standard/sensors/external_task.py index 9912fe96d70d0..7506a88ec0c81 100644 --- a/providers/standard/src/airflow/providers/standard/sensors/external_task.py +++ b/providers/standard/src/airflow/providers/standard/sensors/external_task.py @@ -59,7 +59,7 @@ if AIRFLOW_V_3_0_PLUS: from airflow.sdk.definitions.context import Context else: - from airflow.utils.context import Context # type: ignore[no-redef] + from airflow.utils.context import Context class ExternalDagLink(BaseOperatorLink): diff --git a/providers/standard/src/airflow/providers/standard/triggers/external_task.py b/providers/standard/src/airflow/providers/standard/triggers/external_task.py index fe90c14fabd8a..0085bb33b866b 100644 --- a/providers/standard/src/airflow/providers/standard/triggers/external_task.py +++ b/providers/standard/src/airflow/providers/standard/triggers/external_task.py @@ -123,7 +123,7 @@ async def run(self) -> typing.AsyncIterator[TriggerEvent]: return allowed_count = await get_count_func(self.allowed_states) - if allowed_count == len(run_id_or_dates): # type: ignore[arg-type] + if allowed_count == len(run_id_or_dates): yield TriggerEvent({"status": "success"}) return self.log.info("Sleeping for %s seconds", self.poke_interval) @@ -140,7 +140,7 @@ async def _get_count_af_3(self, states): } if self.external_task_ids: count = await sync_to_async(RuntimeTaskInstance.get_ti_count)( - task_ids=self.external_task_ids, # type: ignore[arg-type] + task_ids=self.external_task_ids, states=states, **params, ) @@ -233,7 +233,7 @@ async def run(self) -> typing.AsyncIterator[TriggerEvent]: return else: while True: - num_dags = await self.count_dags() # type: ignore[call-arg] + num_dags = await self.count_dags() if num_dags == runs_ids_or_dates: yield TriggerEvent(self.serialize()) return diff --git a/providers/standard/src/airflow/providers/standard/utils/skipmixin.py b/providers/standard/src/airflow/providers/standard/utils/skipmixin.py index 554c675bfa40f..6be0533fcbbbe 100644 --- a/providers/standard/src/airflow/providers/standard/utils/skipmixin.py +++ b/providers/standard/src/airflow/providers/standard/utils/skipmixin.py @@ -45,7 +45,7 @@ def _ensure_tasks(nodes: Iterable[DAGNode]) -> Sequence[Operator]: from airflow.sdk.definitions.mappedoperator import MappedOperator else: from airflow.models.baseoperator import BaseOperator # type: ignore[no-redef] - from airflow.models.mappedoperator import MappedOperator # type: ignore[no-redef] + from airflow.models.mappedoperator import MappedOperator return [n for n in nodes if isinstance(n, (BaseOperator, MappedOperator))] diff --git a/providers/standard/tests/unit/standard/sensors/test_python.py b/providers/standard/tests/unit/standard/sensors/test_python.py index ff02c6fcbf728..c363af3a9c4c7 100644 --- a/providers/standard/tests/unit/standard/sensors/test_python.py +++ b/providers/standard/tests/unit/standard/sensors/test_python.py @@ -33,7 +33,7 @@ class TestPythonSensor(BasePythonTest): # TODO(kaxil) - this typing hint is fishy - opcls = PythonSensor # type: ignore[assignment] + opcls = PythonSensor def test_python_sensor_true(self): self.run_as_task(fn=lambda: True)