riskcal/setup.py at main · interpretable-dp/riskcal · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
from setuptools import setup, Extension
from setuptools.command.build_ext import build_ext
import os

NO_COMPILE = os.environ.get("RISKCAL_NO_COMPILE") == "1"


class OptionalBuildExt(build_ext):
    """Build Cython extensions, silently falling back to pure Python on failure."""

    def run(self):
        if NO_COMPILE:
            print(
                "riskcal: RISKCAL_NO_COMPILE=1 -> skipping Cython compilation; "
                "using pure Python fallback."
            )
            return
        try:
            super().run()
        except Exception as e:
            print("riskcal: Cython compilation failed; falling back to pure Python.")
            print(f"  {e}")

    def build_extension(self, ext):
        try:
            super().build_extension(ext)
            print(f"riskcal: Cython compiled successfully: {ext.name}")
        except Exception as e:
            print(
                f"riskcal: WARNING: Cython compilation failed; "
                f"using pure Python for {ext.name}"
            )
            print(f"  {e}")


def get_extensions():
    if NO_COMPILE:
        return []

    try:
        from Cython.Build import cythonize
        import numpy as np
    except Exception as e:
        print("riskcal: Cython/numpy not available for build; using pure Python.")
        print(f"  {e}")
        return []

    ext = Extension(
        name="riskcal.analysis.rdp.converter",
        sources=["src/riskcal/analysis/rdp/converter.pyx"],
        include_dirs=[np.get_include()],
    )

    return cythonize(
        [ext],
        compiler_directives={"language_level": "3"},
        annotate=False,
    )


setup(
    package_dir={"": "src"},
    ext_modules=get_extensions(),
    cmdclass={"build_ext": OptionalBuildExt},
)