aboutsummaryrefslogtreecommitdiffstats
path: root/build_scripts/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'build_scripts/utils.py')
-rw-r--r--build_scripts/utils.py10
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