fix cython setup

AWehrhahn · Jun 8, 2021 · 791a14f · 791a14f
1 parent be98d42
commit 791a14f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cython/setup.py b/cython/setup.py
@@ -2,6 +2,7 @@
 from distutils.core import setup
 from distutils.extension import Extension
 from Cython.Build import cythonize
+import numpy as np
 
 lib_path = realpath(join(dirname(__file__), "../lib"))
 file_path = realpath(join(dirname(__file__), "smelib.pyx"))
@@ -12,6 +13,7 @@
     libraries=["gfortran", "sme"],
     library_dirs=[lib_path],
     runtime_library_dirs=[lib_path],
+    include_dirs=[np.get_include()],
 )
 setup(
     name="smelib",