Merge branch 'master' into duckdb-integration

flyteorg · Feb 23, 2023 · 9a0e42b · 9a0e42b
2 parents 198bafb + b3ad158
commit 9a0e42b
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/flytekit/core/map_task.py b/flytekit/core/map_task.py
@@ -221,7 +221,12 @@ def _raw_execute(self, **kwargs) -> Any:
         return outputs
 
 
-def map_task(task_function: PythonFunctionTask, concurrency: int = 0, min_success_ratio: float = 1.0, **kwargs):
+def map_task(
+    task_function: typing.Union[typing.Callable, PythonFunctionTask],
+    concurrency: int = 0,
+    min_success_ratio: float = 1.0,
+    **kwargs,
+):
     """
     Use a map task for parallelizable tasks that run across a list of an input type. A map task can be composed of
     any individual :py:class:`flytekit.PythonFunctionTask`.

diff --git a/flytekit/py.typed b/flytekit/py.typed
diff --git a/plugins/setup.py b/plugins/setup.py
@@ -82,4 +82,5 @@ def run(self):
     classifiers=["Private :: Do Not Upload to pypi server"],
     install_requires=[],
     cmdclass={"install": InstallCmd, "develop": DevelopCmd},
+    package_data={"flytekit": ["py.typed"]},
 )