setup.py

from setuptools import setup, find_packages
import sys
import os

# PEP0440 compatible formatted version, see:
# https://www.python.org/dev/peps/pep-0440/
#
# release markers:
#   X.Y
#   X.Y.Z   # For bugfix releases
#
# pre-release markers:
#   X.YaN   # Alpha release
#   X.YbN   # Beta release
#   X.YrcN  # Release Candidate
#   X.Y     # Final release

# version.py defines the VERSION and VERSION_SHORT variables.
# We use exec here so we don't import allennlp_hub whilst setting up.
VERSION = {}
with open("allennlp_hub/version.py") as version_file:
    exec(version_file.read(), VERSION)

install_requirements = []
if not os.environ.get("EXCLUDE_ALLENNLP_IN_SETUP"):
    # Warning: This will not give you the desired version if you've already
    # installed allennlp! See https://github.com/pypa/pip/issues/5898.
    #
    # There used to be an alternative to this using `dependency_links`
    # (https://stackoverflow.com/questions/3472430), but pip decided to
    # remove this in version 19 breaking numerous projects in the process.
    # See https://github.com/pypa/pip/issues/6162.
    #
    # As a mitigation, run `pip uninstall allennlp` before installing this
    # package.
    #
    # TODO(brendanr): Make these point to released versions, when possible, by
    # loading requirements.txt. Currently allennlp-semparse is unreleased and
    # it depends on a specific allennlp SHA. Due to the aforementioned
    # setuptools bug, we explicitly set the allennlp version here to be that
    # required by allennlp-semparse.
    allennlp_sha = "65ff0d87a30e6532cb21ea5fe8b7bd436445c128"
    semparse_sha = "339e617861a7616618a503cc98e1e9c8b28a1b06"
    reading_comprehension_sha = "e4f8e5df4f9fa35287d44e94fc8b26b9cabed0a5"
    install_requirements = [
        f"allennlp @ git+https://github.com/allenai/allennlp@{allennlp_sha}#egg=allennlp",
        f"allennlp_semparse @ git+https://github.com/allenai/allennlp-semparse@{semparse_sha}#egg=allennlp-semparse",
        f"allennlp_reading_comprehension @ git+https://github.com/allenai/allennlp-reading-comprehension@{reading_comprehension_sha}#egg=allennlp-reading-comprehension",
    ]

# make pytest-runner a conditional requirement,
# per: https://github.com/pytest-dev/pytest-runner#considerations
needs_pytest = {"pytest", "test", "ptr"}.intersection(sys.argv)
pytest_runner = ["pytest-runner"] if needs_pytest else []

setup_requirements = [
    # add other setup requirements as necessary
] + pytest_runner

setup(
    name="allennlp_hub",
    version=VERSION["VERSION"],
    description="A collection of selected of models built with AllenNLP.",
    long_description=open("README.md").read(),
    long_description_content_type="text/markdown",
    classifiers=[
        "Intended Audience :: Science/Research",
        "Development Status :: 3 - Alpha",
        "License :: OSI Approved :: Apache Software License",
        "Programming Language :: Python :: 3.6",
        "Topic :: Scientific/Engineering :: Artificial Intelligence",
    ],
    keywords="allennlp NLP deep learning machine reading",
    url="https://github.com/allenai/allennlp-hub",
    author="Allen Institute for Artificial Intelligence",
    author_email="allennlp@allenai.org",
    license="Apache",
    packages=find_packages(exclude=["*.tests", "*.tests.*", "tests.*", "tests"]),
    install_requires=install_requirements,
    setup_requires=setup_requirements,
    tests_require=["pytest", "flaky", "responses>=0.7"],
    include_package_data=True,
    python_requires=">=3.6.1",
    zip_safe=False,
)