[stubs] install alongside implementation

srittau · Oct 22, 2017 · ec9f68a · ec9f68a
1 parent 0b93433
commit ec9f68a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -2,6 +2,7 @@
 
 import os
 from setuptools import setup
+import sys
 
 
 def read(fname):
@@ -17,6 +18,10 @@ def read(fname):
     author_email="srittau@rittau.biz",
     url="https://github.com/srittau/python-asserts",
     py_modules=["asserts", "test_asserts"],
+    data_files=[(
+        "lib/python{}.{}/site-packages".format(*sys.version_info[:2]),
+        ["asserts.pyi"],
+    )],
     license="MIT",
     classifiers=[
         "Development Status :: 3 - Alpha",