numpy_.py

"""NumPy helper.

Note: If you plan to add a library detection script like this one, consider it twice. Most library detection should go
to CMake script. This one is an exception, because Python code can do a much better job due to NumPy's inherent Pythonic
nature.
"""

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()