Revert "Make flytekit comply with PEP-561 (#1516)" (#1532)

This reverts commit b3ad158. Signed-off-by: Eduardo Apolinario <[email protected]> Co-authored-by: Eduardo Apolinario <[email protected]>
flyteorg · Mar 3, 2023 · 8ccb5dd · 8ccb5dd
1 parent 71d436a
commit 8ccb5dd
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 7 deletions.
diff --git a/flytekit/core/map_task.py b/flytekit/core/map_task.py
@@ -221,12 +221,7 @@ def _raw_execute(self, **kwargs) -> Any:
         return outputs
 
 
-def map_task(
-    task_function: typing.Union[typing.Callable, PythonFunctionTask],
-    concurrency: int = 0,
-    min_success_ratio: float = 1.0,
-    **kwargs,
-):
+def map_task(task_function: 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
@@ -84,5 +84,4 @@ def run(self):
     classifiers=["Private :: Do Not Upload to pypi server"],
     install_requires=[],
     cmdclass={"install": InstallCmd, "develop": DevelopCmd},
-    package_data={"flytekit": ["py.typed"]},
 )