setup.py

import os


import setuptools

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

"""
Load data from the__versions__.py module. Change version, etc in
that module, and it will be automatically populated here. This allows us to
access the module version, etc from inside python with

Examples:

    >>> from wellcomeml.common import about
    >>> about['__version__']
    2019.10.0

"""

about = {}  # type: dict
version_path = os.path.join(here, 'wellcomeml', '__version__.py')
with open(version_path, 'r') as f:
    exec(f.read(), about)

with open('README.md', 'r') as f:
    long_description = f.read()

extras = {
        'core': [
            'scikit-learn',
            'scipy',
            'umap-learn',
            'gensim',
            'bokeh',
            'pandas',
            'nervaluate'
        ],
        'transformers': [
            'transformers',
            'tokenizers'
        ],
        'tensorflow': [
            'tensorflow==2.4.0',
            'tensorflow-addons',
            'numpy>=1.19.2,<1.20'
        ],
        'torch': [
            'torch'
        ],
        'spacy': [
            'spacy[lookups]==3.0.6',
            'click>=7.0,<8.0'
        ],
}

# Allow users to install 'all' if they wish
extras['all'] = [dep for dep_list in extras.values() for dep in dep_list]

setuptools.setup(
    name=about['__name__'],
    version=about['__version__'],
    author=about['__author__'],
    author_email=about['__author_email__'],
    description=about['__description__'].replace('\n', ''),
    long_description=long_description,
    long_description_content_type='text/markdown',
    url=about['__url__'],
    license=about['__license__'],
    packages=setuptools.find_packages(include=["wellcomeml*"]),
    classifiers=[
        'Programming Language :: Python :: 3',
        'Operating System :: OS Independent',
    ],
    install_requires=[
        'boto3',
        'twine',
        'cython',
        'tqdm'
    ],
    extras_require=extras,
    tests_require=[
        'pytest',
        'flake8',
        'black',
        'pytest-cov'
        'tox'
    ]
)