myCherryML/setup.py at main · elileka/myCherryML · 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
from setuptools import setup, find_packages
from setuptools import setup, Extension
from Cython.Build import cythonize
import numpy as np


def get_version_from_init(init_path='cherryml/__init__.py'):
    with open(init_path, 'r') as f:
        for line in f:
            if line.startswith('__version__'):
                return eval(line.split('=')[-1])


version = get_version_from_init()


extensions = [
    Extension(
        "cherryml._siterm.fast_site_rates",
        ["cherryml/_siterm/fast_site_rates.pyx"],
        include_dirs=[np.get_include()],
        language="c++",  # Tell the compiler to use C++
        extra_compile_args=["-std=c++11", "-O3"],  # Optional: Use C++11 standard
    )
]


setup(
    name='cherryml',
    version=version,
    packages=find_packages(),
    include_package_data=True,
    install_requires=[
        'biotite',
        'black',
        'ete3',
        'flake8',
        'isort',
        'matplotlib',
        'networkx',
        'numpy',
        'pandas',
        'parameterized',
        'pytest',
        'scipy',
        'seaborn',
        'threadpoolctl',
        'torch',
        'tqdm',
        'wget',
    ],
    ext_modules=cythonize(extensions),
)