correct setup.py to include numpy headers

07abe660 · Mozul Rémy · 844bf463 · 07abe660
Commit 07abe660 authored Apr 03, 2014 by Mozul Rémy
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

setup.py setup.py +5 -1

No files found.
--- a/setup.py
+++ b/setup.py
 import os
+import numpy
 from distutils.core import setup
 from distutils.extension import Extension
 from Cython.Distutils import build_ext
 setup(name='sprint',
-      ext_modules=[Extension('test',['test.pyx'],libraries=['type'],library_dirs=[os.getcwd()])],
+      ext_modules=[Extension('test',['test.pyx'],
+                             libraries=['type'],
+                             library_dirs=[os.getcwd()],
+                             include_dirs=[numpy.get_include()])],
      cmdclass={'build_ext':build_ext}
      )