setup.py

from setuptools import setup, Extension, find_packages
from Cython.Build import cythonize
import numpy
import os

def build_ext(srcs, package="puffergrid"):
    return Extension(
        name=package + "." + srcs[0].split('/')[-1].split('.')[0],
        sources=srcs,
        define_macros=[('NPY_NO_DEPRECATED_API', 'NPY_1_7_API_VERSION')],
        include_dirs=[numpy.get_include()],
    )

ext_modules = [
    build_ext(["puffergrid/action.pyx"]),
    build_ext(["puffergrid/event.pyx"]),
    build_ext(["puffergrid/grid.cpp"]),
    build_ext(["puffergrid/grid_env.pyx"]),
    build_ext(["puffergrid/grid_object.pyx"]),
    build_ext(["puffergrid/observation_encoder.pyx"]),
    build_ext(["puffergrid/stats_tracker.pyx"]),
    build_ext(["examples/forage.pyx"], "puffergrid.examples"),

    build_ext(["tests/test_grid_object.pyx"], "puffergrid.tests"),
    build_ext(["tests/test_action_handler.pyx"], "puffergrid.tests"),
]

debug = os.getenv('DEBUG', '0') == '1'
annotate = os.getenv('ANNOTATE', '0') == '1'
build_dir = 'build'
if debug:
    build_dir = 'build_debug'

os.makedirs(build_dir, exist_ok=True)
os.makedirs("puffergrid/tests", exist_ok=True)
os.makedirs("puffergrid/examples", exist_ok=True)

setup(
    name='puffergrid',
    packages=find_packages(),
    ext_modules=cythonize(
        ext_modules,
        build_dir=build_dir,
        compiler_directives={
            "profile": True,
            "language_level": "3",
            "embedsignature": debug,
            "annotation_typing": debug,
            "cdivision": debug,
            "boundscheck": debug,
            "wraparound": debug,
            "initializedcheck": debug,
            "nonecheck": debug,
            "overflowcheck": debug,
            "overflowcheck.fold": debug,
            "linetrace": debug,
            "c_string_encoding": "utf-8",
            "c_string_type": "str",

        },
        annotate=debug or annotate,
    ),
    description='',
    url='https://github.com/daveey/puffergrid',
    install_requires=[
        'numpy',
        'cython==3.0.11',
        'tqdm',
    ],
)