setup.py

from setuptools import setup

install_requires = [
    "huggingface_hub>=0.21",
    "pyyaml~=6.0",
    "wasabi",
    "numpy<2.0",
    "cloudpickle>=1.6",
    "stable-baselines3",
    "moviepy",
]

extras = {}

extras["quality"] = [
    "black~=22.0",
    "isort>=5.5.4",
    "flake8>=3.8.3",
]

extras["test"] = ["pytest", "gymnasium[classic-control]"]

extras["dev"] = extras["quality"] + extras["test"]

setup(
    name="huggingface_sb3",
    version="3.1",
    packages=["huggingface_sb3"],
    url="https://github.com/huggingface/huggingface_sb3",
    license="Apache",
    author="Thomas Simonini, Omar Sanseviero and Hugging Face Team",
    author_email="thomas.simonini@huggingface.co",
    description="Additional code for Stable-baselines3 to load and upload models from the Hub.",
    install_requires=install_requires,
    extras_require=extras,
    long_description=open("README.md", "r", encoding="utf-8").read(),
    long_description_content_type="text/markdown",
    keywords="reinforcement learning deep reinforcement learning RL",
)