setup.py

"""Setup script of rllib."""
from setuptools import find_packages, setup

extras = {
    "test": [
        "pytest>=5.0,<5.1",
        "flake8>=3.7.8,<3.8",
        "pydocstyle==4.0.0",
        "black>=19.10b0",
        "isort>=5.0.0",
        "pytest_cov>=2.7,<3",
        "mypy>=0.750",
    ],
    "envs": ["box2d-py>=2.3.5", "atari_py>=0.2.6", "seaborn>=0.9.0"],
    "mujoco": ["mujoco-py<2.1,>=2.0", "imageio-ffmpeg==0.4.1", "dm_control"],
    "logging": ["tensorboard>=2.0,<3"],
    "experiments": [
        "lsf_runner==0.0.5",
        "torchvision>=0.6.0",
        "Pillow==5.4.1",
        "pandas==0.25.0",
        "dotmap>=1.3.0,<1.4.0",
        "pyyaml>=5.0.0",
        "argparse-dataclass>=0.2.1",
    ],
}
extras["all"] = [item for group in extras.values() for item in group]

setup(
    name="rllib",
    version="0.0.1",
    author="Sebastian Curi",
    author_email="sebascuri@gmail.com",
    license="MIT",
    python_requires=">=3.7.0",
    packages=find_packages(exclude=["docs"]),
    package_data={"rllib": ["environment/mujoco/assets/*.xml"]},
    install_requires=[
        "numpy>=1.14,<2",
        "scipy>=1.3.0",
        "torch>=1.10.0",
        "gym>=0.15.4",
        "gym-legacy-toytext>=0.0.1",
        "pygame==2.1.0",
        "tqdm>=4.0.0,<5.0",
        "matplotlib>=3.1.0",
        "gpytorch>=1.1.1,<1.2.0",
        "tensorboardX>=2.0,<3",
        "seaborn>=0.11,<1",
    ],
    extras_require=extras,
    classifiers=[
        # How mature is this project? Common values are
        #   3 - Alpha
        #   4 - Beta
        #   5 - Production/Stable
        "Development Status :: 3 - Alpha",
        # Indicate who your project is intended for
        "Intended Audience :: Developers",
        "Topic :: Software Development :: Build Tools",
        "License :: OSI Approved :: MIT License",
        "Programming Language :: Python :: 3.7",
    ],
)