Adding np include dirs for cython

nicholasturner1 · Mar 18, 2018 · 79f9ecc · 79f9ecc
1 parent 161b8d0
commit 79f9ecc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -1,10 +1,13 @@
 #!/usr/bin/env python3
 
+import numpy as np
 from Cython.Build import cythonize
 from setuptools import setup, find_packages, Extension
 
 extensions = [
-    Extension("synaptor.seg_utils._seg_utils", sources=["synaptor/seg_utils/_seg_utils.pyx"])
+    Extension("synaptor.seg_utils._seg_utils", 
+              sources=["synaptor/seg_utils/_seg_utils.pyx"],
+              include_dirs=[np.get_include()])
 ]
 
 setup(