"NumPy helper." | |
from .env import check_negative_env_flag | |
# Set USE_NUMPY to what the user wants, because even if we fail here, cmake | |
# will check for the presence of NumPy again (`cmake/Dependencies.cmake`). | |
USE_NUMPY = not check_negative_env_flag('USE_NUMPY') | |
NUMPY_INCLUDE_DIR = None | |
if USE_NUMPY: | |
try: | |
import numpy as np | |
except ImportError: | |
pass | |
else: | |
# To reach here, the user must has not disabled NumPy build and the | |
# NumPy library is present in the system. | |
NUMPY_INCLUDE_DIR = np.get_include() |