ranvar/setup.py at main · ajvogel/ranvar · 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
# # -*- coding: utf-8 -*-
# from setuptools import setup

# packages = \
# ['casino']

# package_data = \
# {'': ['*']}

# install_requires = \
# ['cython>=0.29.35,<0.30.0', 'numba>=0.57.0,<0.58.0', 'numpy>=1.24.3,<2.0.0']

# setup_kwargs = {
#     'name': 'casino',
#     'version': '0.1.0',
#     'description': 'Casino enables fast probabilistic modelling',
#     'long_description': '# Casino Probabilistic Modelling Library\n\nThis is awesome.',
#     'author': 'Adolph Vogel',
#     'author_email': 'ajvogel@gmail.com',
#     'maintainer': 'None',
#     'maintainer_email': 'None',
#     'url': 'None',
#     'packages': packages,
#     'package_data': package_data,
#     'install_requires': install_requires,
#     'python_requires': '>=3.11,<3.12',
# }
# from build import *
# build(setup_kwargs)

# setup(**setup_kwargs)


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

extensions = [
    Extension("ranvar.random", ["ranvar/random.py"]),
    Extension("ranvar.vm", ["ranvar/vm.py"]),
    Extension("ranvar.digest", ["ranvar/digest.py"], include_dirs=[np.get_include()]),
    Extension(
        "ranvar.cdigest",
        sources=["ranvar/cdigest.pyx"],
        include_dirs=[np.get_include(), "ranvar/cpp"],
        language="c++",
        extra_compile_args=["-std=c++17"],
    ),
]

setup(
    name="ranvar",
    ext_modules=cythonize(
        extensions,
        include_path=["ranvar/", np.get_include()],
        force=True,
        annotate=True,
    ),
)