blob: b5f32cd1f7a6ed4c95303326b8f8c1fc2f286b60 [file] [log] [blame]
"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()