You've already forked STARK
mirror of
https://github.com/MarkParker5/STARK.git
synced 2026-04-28 19:34:24 +02:00
Mark Parker
24 lines
622 B
Python
24 lines
622 B
Python
import glob
|
|
import os
|
|
import sys
|
|
from setuptools import Extension, setup
|
|
from Cython.Build import cythonize
|
|
import numpy
|
|
|
|
ext_modules = [
|
|
Extension(path.replace("/", ".").replace(".pyx", ""), [path])
|
|
for path in glob.glob("**/*.pyx", recursive=True)
|
|
]
|
|
|
|
# if force in args, cleanup
|
|
if "--force" in sys.argv:
|
|
extensions = [".so", ".dll", ".tmp", ".c"]
|
|
for ext in extensions:
|
|
for file_path in glob.glob(f"**/*{ext}", recursive=True):
|
|
os.remove(file_path)
|
|
|
|
setup(
|
|
ext_modules=cythonize(ext_modules, compiler_directives={"boundscheck": False}),
|
|
include_dirs=[numpy.get_include()],
|
|
)
|