diff options
Diffstat (limited to 'build_scripts/utils.py')
-rw-r--r-- | build_scripts/utils.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/build_scripts/utils.py b/build_scripts/utils.py index d80e59a3d..0a9036905 100644 --- a/build_scripts/utils.py +++ b/build_scripts/utils.py @@ -38,6 +38,7 @@ ############################################################################# import sys +from pathlib import Path import os import re import stat @@ -83,6 +84,15 @@ def update_env_path(newpaths): os.environ['PATH'] = f"{path}{os.pathsep}{os.environ['PATH']}" +def get_numpy_location(): + for p in sys.path: + if 'site-' in p: + numpy = Path(p).resolve() / 'numpy' + if numpy.is_dir(): + return os.fspath(numpy / 'core' / 'include') + return None + + def winsdk_setenv(platform_arch, build_type): from distutils.msvc9compiler import VERSION as MSVC_VERSION from distutils.msvc9compiler import Reg |