From d95723e260ceda8a779a831e8ed515ca76f0e272 Mon Sep 17 00:00:00 2001
From: Yicheng Luo <ethanluoyc@gmail.com>
Date: Wed, 1 Nov 2023 13:38:18 +0000
Subject: [PATCH] Fix packaging issue

---
 MANIFEST.in | 2 ++
 setup.py    | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/MANIFEST.in b/MANIFEST.in
index 4cc82c5dd..365d65ff3 100644
--- a/MANIFEST.in
+++ b/MANIFEST.in
@@ -4,3 +4,5 @@ include requirements.txt
 include requirements-jax.txt
 include requirements-test.txt
 include haiku/py.typed
+
+global-exclude *_test.py
diff --git a/setup.py b/setup.py
index 25b39b183..7fd8c5ad9 100644
--- a/setup.py
+++ b/setup.py
@@ -46,7 +46,7 @@ def _parse_requirements(requirements_txt_path):
     long_description_content_type='text/markdown',
     author_email='haiku-dev-os@google.com',
     # Contained modules and scripts.
-    packages=find_namespace_packages(exclude=['*_test.py', 'examples']),
+    packages=find_namespace_packages(exclude=['examples*', 'docs*']),
     install_requires=_parse_requirements('requirements.txt'),
     extras_require={'jax': _parse_requirements('requirements-jax.txt')},
     tests_require=_parse_requirements('requirements-test.txt'),