Fix TaskDecorator type hints (#21881)

apache · Mar 1, 2022 · 5befc7f · 5befc7f
1 parent 3035d3a
commit 5befc7f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/airflow/decorators/__init__.pyi b/airflow/decorators/__init__.pyi
@@ -20,7 +20,7 @@
 # necessarily exist at run time. See "Creating Custom @task Decorators"
 # documentation for more details.
 
-from typing import Any, Dict, Iterable, List, Mapping, Optional, Union, overload
+from typing import Any, Callable, Dict, Iterable, List, Mapping, Optional, Union, overload
 
 from airflow.decorators.base import Function, Task, TaskDecorator
 from airflow.decorators.branch_python import branch_task

diff --git a/airflow/decorators/base.py b/airflow/decorators/base.py
@@ -476,7 +476,7 @@ def __call__(self, python_callable: Function) -> Task[Function]:
     def __call__(
         self,
         *,
-        multiple_outputs: Optional[bool],
+        multiple_outputs: Optional[bool] = None,
         **kwargs: Any,
     ) -> Callable[[Function], Task[Function]]:
         """For the decorator factory ``@task()`` case."""