setup.py

from os import path
from setuptools import setup, find_packages

this_dir = path.abspath(path.dirname(__file__))
with open(
    path.join(this_dir, "utils", "setup", "README.pypi.md"), encoding="utf-8"
) as f:
    long_description = f.read()

setup(
    name="smarts",
    description="Scalable Multi-Agent RL Training School",
    long_description=long_description,
    long_description_content_type="text/markdown",
    version="0.4.18",
    packages=find_packages(exclude=("tests", "examples")),
    include_package_data=True,
    zip_safe=True,
    python_requires=">=3.7",
    install_requires=[
        # setuptools:
        #   tensorboard needs >=41
        #   50.0 is broken: https://github.com/pypa/setupatools/issues/2353
        "setuptools>=41.0.0,!=50.0",
        "cached-property",
        "click",  # used in scl
        "gym",
        "numpy",
        "rich",
        "rtree",  # Used by sumolib
        "pandas",
        "psutil",
        "visdom",
        "pybullet==3.0.6",
        "sklearn",  # KDTree from sklearn is used by waypoints
        "tableprint",
        "trimesh",  # Used for writing .glb files
        "pynput",  # Used by HumanKeyboardAgent
        "sh",
        "shapely",
        # The following are for Scenario Studio
        "yattag",
        # The following are for /envision
        "cloudpickle<1.4.0",
        "tornado",
        "websocket-client",
        # The following are for the /smarts/algorithms
        "matplotlib",
        "scikit-image",
        # The following are for /smarts/zoo
        "grpcio==1.37.0",
        "PyYAML",
        "twisted",
        # The following are used for imitation learning
        "ijson",
    ],
    extras_require={
        "test": [
            # The following are for testing
            "ipykernel",
            "pytest",
            "pytest-benchmark",
            "pytest-cov",
            "pytest-notebook",
            "pytest-xdist",
            "ray[rllib]==1.8.0",  # We use Ray for our multiprocessing needs
        ],
        "train": [
            "tensorflow==2.2.1",
            # XXX: TF requires specific version of scipy
            "scipy==1.4.1",
            "torch==1.4.0",
            "torchvision==0.5.0",
            "ray[rllib]==1.8.0",  # We use Ray for our multiprocessing needs
        ],
        "dev": [
            "black==20.8b1",
            "grpcio-tools==1.37.0",
            "isort==5.7.0",
            "sphinx",
            "sphinx-rtd-theme",
            "sphinxcontrib-apidoc",
        ],
        "camera-obs": [
            "panda3d",
            "panda3d-gltf",
        ],
    },
    entry_points={"console_scripts": ["scl=cli.cli:scl"]},
)