diff options
Diffstat (limited to 'numpy/lib/utils.py')
| -rw-r--r-- | numpy/lib/utils.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/numpy/lib/utils.py b/numpy/lib/utils.py index 06c1252eb..5e50ba487 100644 --- a/numpy/lib/utils.py +++ b/numpy/lib/utils.py @@ -34,7 +34,7 @@ def get_include(): assert len(include_dirs)==1,`include_dirs` return include_dirs[0] -def get_numarray_include(): +def get_numarray_include(type=None): """Return the directory in the package that contains the numpy/*.h header files. @@ -46,8 +46,12 @@ def get_numarray_include(): include_dirs=[numpy.get_include()]) """ from numpy.numarray import get_numarray_include_dirs + from numpy.distutils.misc_util import get_numpy_include_dirs include_dirs = get_numarray_include_dirs() - return include_dirs[0] + if type is None: + return include_dirs[0] + else: + return include_dirs + get_numpy_include_dirs() # Adapted from Albert Strasheim |