setup.py

from setuptools import setup
from setuptools.extension import Extension
import numpy

USE_CYTHON = True
ext = '.pyx' if USE_CYTHON else '.c'

extensions = [Extension("propagators.contfrac", ["propagators/contfrac"+ext], include_dirs=['.', numpy.get_include()]),
              Extension("propagators._cn_utils", ["propagators/_cn_utils"+ext], include_dirs=['.', numpy.get_include()]),
              Extension("propagators.dispersion_relations", ["propagators/dispersion_relations"+ext], include_dirs=['.'])
              ]

if USE_CYTHON:
    from Cython.Build import cythonize
    extensions = cythonize(extensions, language_level="3")

setup(
    name='pywaveprop',
    version='1.0.0',
    url='https://github.com/mikelytaev/wave-propagation',
    license='MIT',
    author='Mikhail Lytaev',
    author_email='mikelytaev@gmail.com',
    description='Wave propagation framework',
    setup_requires=[
        'numpy'
    ],
    install_requires=[
        'numpy',
        'scipy',
        'mpmath',
        'matplotlib',
        'cython'
    ],
    ext_modules=extensions,
    zip_safe=False,
    packages=['propagators', 'rwp', 'uwa', 'transforms', 'transforms.fcc_fourier'],
    package_data={
        '': ['*.pyx']
    },
    keywords=["wave propagation", "parabolic equation", "troposphere", "underwater acoustics", "diffraction", "refraction"]
)