setup.py

from setuptools import find_packages, setup
import src.imitation

# TF 1.14.0 is not compatible with sacred because of a TF bug.
TF_VERSION = '>=1.13.1,<2.0,!=1.14.0'
TESTS_REQUIRE = [
    'codecov',
    'codespell',
    'pytest',
    'pytest-cov',
]

setup(
    name='imitation',
    version=src.imitation.__version__,
    description=(
        'Implementation of modern IRL and imitation learning algorithms.'),
    author='Center for Human-Compatible AI and Google',
    python_requires='>=3.6.0',
    packages=find_packages('src'),
    package_dir={'': 'src'},
    package_data={'evaluating_rewards': ['py.typed']},
    install_requires=[
        'awscli',
        'gym',
        'numpy>=1.15',
        'ray[debug]==0.7.4',
        'tqdm',
        'scikit-learn>=0.21.2',
        # TODO(adam): Change to >=2.9.0 once 2.9.0 is released, until then track master
        'stable-baselines @ git+https://github.com/hill-a/stable-baselines.git',
        'jax!=0.1.37',
        'jaxlib~=0.1.20',
        # sacred==0.7.5 build is broken without pymongo
        # sacred>0.7.4 have non-picklable config objects (see GH #109)
        'sacred==0.7.4',
    ],
    tests_require=TESTS_REQUIRE,
    extras_require={
        'gpu': [f'tensorflow-gpu{TF_VERSION}'],
        'cpu': [f'tensorflow{TF_VERSION}'],
        # recommended packages for development
        'dev': [
            'autopep8',
            'flake8',
            'flake8-blind-except',
            'flake8-builtins',
            'flake8-commas',
            'flake8-debugger',
            'flake8-isort',
            'sphinx',
            'sphinxcontrib-napoleon',
            'ipdb',
            'isort',
            'jupyter',
            'pytype',
            # for convenience
            *TESTS_REQUIRE,
        ],
        'test':
        TESTS_REQUIRE,
    },
    entry_points={
        'console_scripts': [
            ('imitation-expert-demos=imitation.scripts.expert_demos'
             ':main_console'),
            'imitation-train=imitation.scripts.train_adversarial:main_console',
        ],
    },
    url='https://github.com/HumanCompatibleAI/imitation',
    license='MIT',
    classifiers=[
        # Trove classifiers
        # Full list: https://pypi.python.org/pypi?%3Aaction=list_classifiers
        'License :: OSI Approved :: MIT License',
        'Programming Language :: Python',
        'Programming Language :: Python :: 3',
        'Programming Language :: Python :: 3.6',
        'Programming Language :: Python :: 3.7',
        'Programming Language :: Python :: Implementation :: CPython',
        'Programming Language :: Python :: Implementation :: PyPy',
    ],
)