setup.py

from pathlib import Path
from typing import Tuple

import numpy as np
from Cython.Build import cythonize
from Cython.Compiler.Options import get_directive_defaults
from setuptools import Extension, setup

_compiler_directives = get_directive_defaults()

CYTHON_PROFILE_MODE = False

CYTHON_MACROS: Tuple[str,str] = None

if CYTHON_PROFILE_MODE:
    _compiler_directives["linetrace"] = True
    _compiler_directives["profile"] = True
    _compiler_directives["emit_code_comments"] = True
    CYTHON_MACROS = [("CYTHON_TRACE", "1")]

_compiler_directives["binding"] = True
_compiler_directives["language_level"] = "3"

extensions = [
    Extension(
        name="ark.utils._bootstrapping",
        sources=[Path( "src", "ark", "utils", "_bootstrapping.pyx").as_posix()],
        include_dirs=[np.get_include()],
        define_macros=CYTHON_MACROS,
    )
]

setup(
    ext_modules=cythonize(extensions, compiler_directives=_compiler_directives),
)