setup.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""This module contains the specification of the mindmeld package"""
# pylint: disable=locally-disabled,invalid-name
from setuptools import setup

with open("README.md") as readme_file:
    readme = readme_file.read()

requirements = [
    'dataclasses<0.8,>=0.7; python_version >= "3.6" and python_version < "3.7"',
    "aiohttp>=3.6.2",
    "attrs>=18.2",  # attrs has a stable API and does not use semver
    "Click~=7.1",
    "click-log==0.1.8",
    "distro~=1.3",
    # elasticsearch-py 7.14 breaks backwards compatibility with servers prior to 7.11
    "elasticsearch>=5.0,<7.14",
    "Flask>=1.1.3,<2",
    "Flask-Cors~=3.0",
    "future~=0.17",
    "pycountry",  # uses calendar versioning
    "nltk~=3.2",
    "numpy~=1.15",
    "pip>=9.0.1",
    "py~=1.4",
    "python-dateutil~=2.6",
    "pytz",  # uses calendar versioning
    "scipy>=0.13.3,<2.0",
    'scikit-learn>=0.18.1,<0.20; python_version < "3.7"',
    'scikit-learn~=1.0.2; python_version >= "3.7"',
    "requests>=2.20.1,<3.0",
    "tqdm~=4.15",
    "immutables~=0.9",
    "pyyaml>=5.1.1",
    "spacy~=2.3,!=2.3.6",  # avoid 2.3.6 because it was yanked from PyPI
    "mypy>=0.782",
    "marshmallow~=3.7.1",
    # We currently depend on an older flask, which requires an older version of jinja, which doesn't pin its version of
    # markupsafe, and markupsafe 2.1.0 removed a deprecated function that the older jinja version depends on.
    # As a stop gap, we'll pin our version of markupsafe to that last 2.0 version. Longer term we need to update
    # our flask dependency, and likely move it to an extra
    "markupsafe==2.0.1",
    'torch~=1.10.0; python_version>="3.6"',
    'pytorch-crf~=0.7.2',
    "protobuf~=3.19.4"
]

setup_requirements = ["pytest-runner~=2.11", "setuptools>=36"]

test_requirements = [
    "flake8==3.5.0",
    "pylint~=2.6.0",
    "pytest==3.8.0",
    "pytest-cov==2.4.0",
    "pytest-asyncio==0.8.0",
    "black>=19.10b0 ; python_version >= '3.6'",
]

setup(
    name="mindmeld",
    version="4.7.0rc2",
    description="A Conversational AI platform.",
    long_description=readme,
    long_description_content_type="text/markdown",
    author="Cisco Systems, Inc.",
    author_email="contact@mindmeld.com",
    url="https://github.com/cisco/mindmeld",
    packages=[
        "mindmeld",
    ],
    package_dir={"mindmeld": "mindmeld"},
    entry_points={"console_scripts": ["mindmeld=mindmeld.cli:cli"]},
    include_package_data=True,
    install_requires=requirements,
    zip_safe=False,
    keywords=["mindmeld", "nlp", "ai", "conversational"],
    classifiers=[
        "Development Status :: 5 - Production/Stable",
        "Intended Audience :: Developers",
        "Natural Language :: English",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.6",
        "Programming Language :: Python :: 3.7",
        "Topic :: Scientific/Engineering :: Artificial Intelligence",
        "Topic :: Software Development :: Libraries :: Application Frameworks",
        "License :: OSI Approved :: Apache Software License",
    ],
    setup_requires=setup_requirements,
    test_suite="tests",
    tests_require=test_requirements,
    extras_require={
        "bot": ["ciscosparkapi", "twilio"],
        "tensorflow": [
            'tensorflow~=1.2; python_version < "3.7"',
            'tensorflow>=1.13.1,<2.0; python_version >= "3.7"',
        ],
        "bert": [  # sentence-transformers
            'torch~=1.10.0; python_version>="3.6"',
            'transformers~=4.15.0; python_version>="3.6"',
            'sentence-transformers~=0.3; python_version>="3.6"',
            # elasticsearch-py 7.14 breaks backwards compatibility with servers prior to 7.11
            'elasticsearch>=7.0,<7.14',
        ],
        "examples": [
            'connexion>=2.7.0; python_version>="3.6"',
        ],
        "augment": [
            'torch~=1.10.0; python_version>="3.6"',
            'transformers~=4.15.0; python_version>="3.6"',
            'sentencepiece==0.1.91'
        ],
        "language_annotator": [
            "google-cloud-translate>=3.0.1",
        ],
        "elasticsearch": [
            # elasticsearch-py 7.14 breaks backwards compatibility with servers prior to 7.11
            "elasticsearch>=5.0,<7.14",
        ],
        "active_learning": [
            "matplotlib~=3.3.1",
        ],
        "dvc": [
            'pygit2>=1.5.0,<1.7; python_version < "3.7"',
            "dvc>=1.8.1"
        ],
        "transformers": [  # huggingface-transformers
            'transformers~=4.15.0; python_version>="3.6"',
        ],
    },
)