Merge pull request #12538 from jakevdp:bundle-pyi

PiperOrigin-RevId: 477453094
jax-ml · Sep 28, 2022 · 0282b4b · 0282b4b
2 parents aafc70d + 6e6fb10
commit 0282b4b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -61,7 +61,7 @@ def generate_proto(source):
     author='JAX team',
     author_email='jax-dev@google.com',
     packages=find_packages(exclude=["examples"]),
-    package_data={'jax': ['py.typed']},
+    package_data={'jax': ['py.typed', "*.pyi", "**/*.pyi"]},
     python_requires='>=3.7',
     install_requires=[
         'absl-py',