setup.py

from io import open
from setuptools import find_packages, setup


extras = {}

extras["torch"] = ["torch"]

extras["serving"] = ["pydantic", "uvicorn", "fastapi"]
extras["all"] = extras["serving"] + ["torch"]

extras["testing"] = ["pytest", "pytest-xdist"]
extras["quality"] = ["black", "isort", "flake8", "mypy"]
extras["docs"] = ["recommonmark", "sphinx", "sphinx-markdown-tables", "sphinx-rtd-theme"]
extras["dev"] = extras["testing"] + extras["quality"] + ["torch"]

setup(
    name="sherlock",
    version="0.2.2",
    author="Christoph Alt,Leonhard Hennig,Marc Hübner,Gabriel Kressin",
    author_email="christoph.alt@posteo.de,marc.huebner@dfki.de,leonhard.hennig@dfki.de,gabriel.kressin@dfki.de",
    description="Sherlock - A state-of-the-art information extraction framework",
    long_description=open("README.md", "r", encoding="utf-8").read(),
    long_description_content_type="text/markdown",
    keywords="neural information extraction named entity recognition relation extraction entity linking event extraction",
    license="Apache",
    url="https://github.com/DFKI-NLP/sherlock",
    packages=find_packages(exclude=["*.tests", "*.tests.*",
                                    "tests.*", "tests"]),
    install_requires=[
        # core
        "spacy>=3.3",
        "transformers>=4.20",
        "seqeval",
        "overrides",
        # can't upgrade to 1.13.1 due to allennlp dependencies. 1.13.1 would fix a critical security vulnerability,
        # see https://github.com/pytorch/pytorch/issues/89855
        "torch>=1.12", # with a more recent CUDA versions, you may need to use
        # `pip install torch==1.12.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116`
        "registrable",
        "tensorboardX",
        "allennlp==2.10.1",
        "allennlp-models==2.10.1",

        # code formatting
        "black",
        "flake8",

        # Microscope
        #"jsonnet>=0.10.0 ; sys.platform != 'win32'",
        "flask",
        "flask_cors",
        "gevent",
    ],
    extras_require=extras,
    include_package_data=True,  # include all files specified in the MANIFEST.in file
    entry_points={
        "console_scripts": []
    },
    python_requires='>=3.8',
    classifiers=[
        "Intended Audience :: Science/Research",
        "License :: OSI Approved :: MIT Software License",
        "Programming Language :: Python :: 3",
        "Topic :: Scientific/Engineering :: Artificial Intelligence",
    ],
)