Include numpy headers

dbaranchuk · Sep 18, 2018 · 9e2b2ea · 9e2b2ea
1 parent d1e9ca4
commit 9e2b2ea
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -1,3 +1,4 @@
+import numpy
 from pprint import pprint
 from urllib.request import urlretrieve
 import sys
@@ -44,7 +45,7 @@ def build_extension(self, ext):
 ext = [Extension(name='.'.join(['ivfhnsw', '_' + name]),
                  sources=[os.path.join('interface', '.'.join([name, 'i']))],
                  swig_opts=['-Iinclude', '-c++'],
-                 include_dirs=['include', os.curdir],
+                 include_dirs=['include', os.curdir, numpy.get_include()],
                  libraries=['ivfhnsw', 'hnswlib', 'faiss', 'gomp', 'lapack',],
                  extra_compile_args=['-std=c++11', '-static'],)
                  for name in names]
@@ -56,7 +57,10 @@ def build_extension(self, ext):
     ext_modules=ext,
     package_dir={'': python_src},
     packages=find_packages(python_src),
-    setup_requires=['pytest-runner'],
+    setup_requires=[
+        'pytest-runner',
+        'numpy',
+    ],
     install_requires=[
         'numpy',
     ],