sds/setup.py at master · hanyas/sds · 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
# python setup.py build_ext --inplace

from distutils.core import setup
from distutils.extension import Extension
from Cython.Build import cythonize
import numpy as np


ext_modules = []

ext_modules.append(
    Extension('sds.cython.hmm_cy',
              sources=['sds/cython/hmm_cy.pyx'],
              language='c++')
)

ext_modules.append(
    Extension('sds.cython.clhmm_cy',
              sources=['sds/cython/clhmm_cy.pyx'],
              language='c++')
)

setup(name='sds',
      version='0.0.1',
      description='Switching dynamical systems for control',
      author='Hany Abdulsamad',
      author_email='hany@robot-learning.de',
      install_requires=['numpy', 'scipy', 'matplotlib',
                        'seaborn', 'sklearn', 'tqdm',
                        'pathos', 'torch', 'cython', 'gym'],
      packages=['sds'], ext_modules=cythonize(ext_modules),
      include_dirs=[np.get_include()],)