setup.py - include type stubs

google · Nov 12, 2024 · 9680040 · 9680040
1 parent 4b94842
commit 9680040
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/python/__init__.py b/python/__init__.py
@@ -0,0 +1 @@
+from .brotli import *
diff --git a/setup.py b/setup.py
@@ -114,8 +114,6 @@ def build_extension(self, ext):
 
 PACKAGE_DIR = {'': 'python'}
 
-PY_MODULES = ['brotli']
-
 class VersionedExtension(Extension):
   def __init__(self, *args, **kwargs):
     define_macros = []
@@ -268,7 +266,8 @@ def get_tag(self):
     platforms=PLATFORMS,
     classifiers=CLASSIFIERS,
     package_dir=PACKAGE_DIR,
-    py_modules=PY_MODULES,
+    packages=[''],
+    package_data={'': ['__init__.py', 'brotli.py', 'brotli.pyi', 'py.typed']},
     ext_modules=EXT_MODULES,
     test_suite=TEST_SUITE,
     cmdclass=CMD_CLASS)