static check fix

Author: amoghrajesh

task-sdk/src/airflow/sdk/definitions/mappedoperator.py

@@ -21,7 +21,7 @@
 import copy
 import warnings
 from collections.abc import Collection, Iterable, Iterator, Mapping, Sequence
-from typing import TYPE_CHECKING, Any, ClassVar
+from typing import TYPE_CHECKING, Any, ClassVar, Union
 
 import attrs
 import methodtools
@@ -51,7 +51,7 @@
 from airflow.sdk.definitions._internal.types import NOTSET
 from airflow.serialization.enums import DagAttributeTypes
 from airflow.task.priority_strategy import PriorityWeightStrategy, validate_and_load_priority_weight_strategy
-from airflow.typing_compat import Literal, TypeAlias, TypeGuard
+from airflow.typing_compat import Literal, TypeGuard
 from airflow.utils.helpers import is_container, prevent_duplicates
 from airflow.utils.xcom import XCOM_RETURN_KEY
 
@@ -79,8 +79,8 @@
     from airflow.utils.task_group import TaskGroup
     from airflow.utils.trigger_rule import TriggerRule
 
-TaskStateChangeCallbackAttrType: TypeAlias = TaskStateChangeCallback | list[TaskStateChangeCallback] | None
-ValidationSource = Literal["expand"] | Literal["partial"]
+TaskStateChangeCallbackAttrType = Union[None, TaskStateChangeCallback, list[TaskStateChangeCallback]]
+ValidationSource = Union[Literal["expand"], Literal["partial"]]
 
 
 def validate_mapping_kwargs(op: type[BaseOperator], func: ValidationSource, value: dict[str, Any]) -> None: