setup.py

import io
import platform
import os
import sys

import setuptools

try:
    from numpy import get_include
except ImportError:
    import subprocess
    errno = subprocess.call([sys.executable, '-m', 'pip', 'install', 'numpy'])
    if errno:
        print('Please install numpy')
        raise SystemExit(errno)
    else:
        from numpy import get_include

try:
    from Cython.Build import cythonize
except ImportError:
    import subprocess
    errno = subprocess.call([sys.executable, '-m', 'pip', 'install', 'Cython'])
    if errno:
        print('Please install Cython')
        raise SystemExit(errno)
    else:
        from Cython.Build import cythonize


# Package meta-data.
NAME = 'river'
DESCRIPTION = 'Online machine learning in Python'
LONG_DESCRIPTION_CONTENT_TYPE = 'text/markdown'
URL = 'https://github.com/online-ml/river'
EMAIL = 'maxhalford25@gmail.com'
AUTHOR = 'Max Halford'
REQUIRES_PYTHON = '>=3.6.0'

# Package requirements.
base_packages = ['numpy>=1.18.1', 'scipy>=1.4.1', 'pandas>=1.0.1']

compat_packages = base_packages + [
    'scikit-learn',
    'scikit-surprise',
    'sqlalchemy',
    'torch',
    'vaex'
]

dev_packages = base_packages + [
    'asv',
    'flake8>=3.7.9',
    'graphviz>=0.10.1',
    'matplotlib>=3.0.2',
    'mypy>=0.761',
    'pytest>=4.5.0',
    'pytest-cov>=2.6.1',
    'pytest-cython>=0.1.0',
    'scikit-learn>=0.22.1',
    'sqlalchemy>=1.3.15'
]

docs_packages = dev_packages + [
    'flask',
    'ipykernel',
    'jupyter-client',
    'mike==0.5.3',
    'mkdocs',
    'mkdocs-awesome-pages-plugin',
    'mkdocs-material',
    'nbconvert',
    'numpydoc'
]

here = os.path.abspath(os.path.dirname(__file__))

# Import the README and use it as the long-description.
with io.open(os.path.join(here, 'README.md'), encoding='utf-8') as f:
    long_description = '\n' + f.read()

# Load the package's __version__.py module as a dictionary.
about = {}
with open(os.path.join(here, NAME, '__version__.py')) as f:
    exec(f.read(), about)

# Where the magic happens:
setuptools.setup(
    name=NAME,
    version=about['__version__'],
    description=DESCRIPTION,
    long_description=long_description,
    long_description_content_type=LONG_DESCRIPTION_CONTENT_TYPE,
    author=AUTHOR,
    author_email=EMAIL,
    python_requires=REQUIRES_PYTHON,
    url=URL,
    packages=setuptools.find_packages(exclude=('tests', 'scikit-multiflow')),
    install_requires=base_packages,
    extras_require={
        'dev': dev_packages,
        'compat': compat_packages,
        'docs': docs_packages
    },
    include_package_data=True,
    license='BSD-3',
    classifiers=[
        # Trove classifiers
        # Full list: https://pypi.python.org/pypi?%3Aaction=list_classifiers
        'License :: OSI Approved :: BSD License',
        'Programming Language :: Python',
        'Programming Language :: Python :: 3',
        'Programming Language :: Python :: 3.6',
        'Programming Language :: Python :: 3.7',
        'Programming Language :: Python :: 3.8',
        'Programming Language :: Python :: Implementation :: CPython',
        'Programming Language :: Python :: Implementation :: PyPy'
    ],
    ext_modules=cythonize(
        module_list=[
            setuptools.Extension(
                '*',
                sources=['**/*.pyx'],
                include_dirs=[get_include()],
                libraries=[] if platform.system() == 'Windows' else ['m']
            )
        ],
        compiler_directives={
            'language_level': 3,
            'binding': True,
            'embedsignature': True
        }
    ) + [setuptools.Extension(
        'river.neighbors.libNearestNeighbor',
        sources=[os.path.join('river', 'neighbors', 'src',
                              'libNearestNeighbor', 'nearestNeighbor.cpp')],
        include_dirs=[get_include()],
        libraries=[] if platform.system() == 'Windows' else ['m'],
        language='c++'
    )]
)