diff --git a/airflow/models/baseoperator.py b/airflow/models/baseoperator.py index 6b33ef9ee392d..0c5019bd58aa6 100644 --- a/airflow/models/baseoperator.py +++ b/airflow/models/baseoperator.py @@ -149,7 +149,7 @@ def _get_parent_defaults(dag: DAG | None, task_group: TaskGroup | None) -> tuple def get_merged_defaults( dag: DAG | None, task_group: TaskGroup | None, - task_params: dict | None, + task_params: collections.abc.MutableMapping | None, task_default_args: dict | None, ) -> tuple[dict, ParamsDict]: args, params = _get_parent_defaults(dag, task_group) @@ -194,7 +194,7 @@ def partial( end_date: datetime | None = None, owner: str = DEFAULT_OWNER, email: None | str | Iterable[str] = None, - params: dict | None = None, + params: collections.abc.MutableMapping | None = None, resources: dict[str, Any] | None = None, trigger_rule: str = DEFAULT_TRIGGER_RULE, depends_on_past: bool = False, @@ -704,7 +704,7 @@ def __init__( wait_for_past_depends_before_skipping: bool = DEFAULT_WAIT_FOR_PAST_DEPENDS_BEFORE_SKIPPING, wait_for_downstream: bool = False, dag: DAG | None = None, - params: dict | None = None, + params: collections.abc.MutableMapping | None = None, default_args: dict | None = None, priority_weight: int = DEFAULT_PRIORITY_WEIGHT, weight_rule: str = DEFAULT_WEIGHT_RULE, diff --git a/airflow/models/dag.py b/airflow/models/dag.py index 53681f6fb616e..61f8b130fa488 100644 --- a/airflow/models/dag.py +++ b/airflow/models/dag.py @@ -18,6 +18,7 @@ from __future__ import annotations import collections +import collections.abc import copy import functools import itertools @@ -403,7 +404,7 @@ def __init__( on_success_callback: None | DagStateChangeCallback | list[DagStateChangeCallback] = None, on_failure_callback: None | DagStateChangeCallback | list[DagStateChangeCallback] = None, doc_md: str | None = None, - params: dict | None = None, + params: collections.abc.MutableMapping | None = None, access_control: dict | None = None, is_paused_upon_creation: bool | None = None, jinja_environment_kwargs: dict | None = None, @@ -3489,7 +3490,7 @@ def dag( on_success_callback: None | DagStateChangeCallback | list[DagStateChangeCallback] = None, on_failure_callback: None | DagStateChangeCallback | list[DagStateChangeCallback] = None, doc_md: str | None = None, - params: dict | None = None, + params: collections.abc.MutableMapping | None = None, access_control: dict | None = None, is_paused_upon_creation: bool | None = None, jinja_environment_kwargs: dict | None = None, diff --git a/airflow/models/param.py b/airflow/models/param.py index 03f6ff414c147..80da2b858e5b5 100644 --- a/airflow/models/param.py +++ b/airflow/models/param.py @@ -165,7 +165,7 @@ class ParamsDict(MutableMapping[str, Any]): __version__: ClassVar[int] = 1 __slots__ = ["__dict", "suppress_exception"] - def __init__(self, dict_obj: dict | None = None, suppress_exception: bool = False): + def __init__(self, dict_obj: MutableMapping | None = None, suppress_exception: bool = False): """ :param dict_obj: A dict or dict like object to init ParamsDict :param suppress_exception: Flag to suppress value exceptions while initializing the ParamsDict