diff --git a/flytekit/core/map_task.py b/flytekit/core/map_task.py
index f888638965..48d0f0b335 100644
--- 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
deleted file mode 100644
index e69de29bb2..0000000000
diff --git a/plugins/setup.py b/plugins/setup.py
index 843e2528bb..1b47cc58e0 100644
--- 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"]},
 )