setup.py

from setuptools import Extension, setup

import numpy

__version__ = "1.2.3"

setup(
    ext_modules=[
        Extension(
            "mt2._mt2",
            ["src/_mt2/main.cpp"],
            define_macros=[
                # Pass in the version info so we can expose it in the extension.
                ("VERSION_INFO", __version__),
                # For reasons explained in lester_mt2_bisect_v7.h, we need to manually
                # enable some inlining optimisations.
                ("ENABLE_INLINING", "1"),
                # Copyright printing is disabled here, since we include the necessary
                # citation information elsewhere.
                ("DISABLE_COPYRIGHT_PRINTING", "1"),
            ],
            include_dirs=[numpy.get_include()],
            language="c++",
            extra_compile_args=["-std=c++11"],
        ),
    ],
)