setup.py

#!/usr/bin/env python
"""pymctdh setup
"""
import os
import sys
from setuptools import setup, Extension, find_packages
from Cython.Build import cythonize
from Cython.Distutils import build_ext

#local_path = os.path.dirname(os.path.abspath(sys.argv[0]))
#os.chdir(local_path)
#sys.path.insert(0, local_path)
#sys.path.insert(0, os.path.join(local_path, 'pymctdh'))  # to retrive _version

ext_modules = [
    Extension("pymctdh.cy.wftools",
        sources=["pymctdh/cy/wftools.pyx"],
        include_dirs = [numpy.get_include(),scipy.get_include()],
        extra_compile_args = ["-O3", "-funroll-loops"],
    ),
    Extension("pymctdh.cy.tensorutils",
        ["pymctdh/cy/tensorutils.pyx"],
        include_dirs = [numpy.get_include()],
        extra_compile_args = ["-O3", "-funroll-loops"],
    )
]

# Setup commands go here
setup(name='pymctdh',
      python_requires='>=3.6',
      requires = ['numpy (>=1.12)', 'scipy (>=1.0)', 'cython (>=0.21)'],
      version='0.1',
      packages=find_packages(),
      include_package_data=True,
      ext_modules = cythonize(EXT_MODULES),
      cmdclass = {'build_ext': build_ext},
      author = 'Addison J. Schile',
      author_email = 'addschile@gmail.com',
      license = 'MIT',
      description = 'an open-source MCTDH package written in python',
      long_description = '',
      keywords = 'quantum dynamics, variational principles, tree tensor networks',
      url = 'https://github.com/addschile/pymctdh'
      # TODO add classifiers
      #classifiers = CLASSIFIERS,
)