Fix InletEventsAccessors type stub (apache#41572)

hussein-awala · Aug 20, 2024 · 2545918 · 2545918
1 parent 716c430
commit 2545918
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/airflow/utils/context.pyi b/airflow/utils/context.pyi
@@ -86,7 +86,7 @@ class InletEventsAccessors(Mapping[str, InletEventsAccessor]):
     def __init__(self, inlets: list, *, session: Session) -> None: ...
     def __iter__(self) -> Iterator[str]: ...
     def __len__(self) -> int: ...
-    def __getitem__(self, key: int | str | Dataset) -> InletEventsAccessor: ...
+    def __getitem__(self, key: int | str | Dataset | DatasetAlias) -> InletEventsAccessor: ...
 
 # NOTE: Please keep this in sync with the following:
 # * KNOWN_CONTEXT_KEYS in airflow/utils/context.py