Simplify dependencies

Author: araffin

.github/workflows/ci.yml

@@ -40,15 +40,11 @@ jobs:
 
         # cpu version of pytorch - faster to download
         pip install torch==2.1.0 --index-url https://download.pytorch.org/whl/cpu
-        pip install pybullet==3.2.5
-        # for v4 MuJoCo envs:
-        pip install mujoco
+
         pip install -r requirements.txt
         # Use headless version
         pip install opencv-python-headless
-        # install parking-env to test HER
-        pip install highway-env==1.8.1
-        pip install -e .
+        pip install -e .[plots,tests]
     - name: Lint with ruff
       run: |
         make lint

.github/workflows/trained_agents.yml

@@ -41,12 +41,10 @@ jobs:
 
         # cpu version of pytorch - faster to download
         pip install torch==2.1.0 --index-url https://download.pytorch.org/whl/cpu
-        pip install pybullet==3.2.5
         pip install -r requirements.txt
         # Use headless version
         pip install opencv-python-headless
-        pip install highway-env==1.8.1
-        pip install -e .
+        pip install -e .[plots,tests]
     - name: Check trained agents
       run: |
         make check-trained-agents

CHANGELOG.md

@@ -9,13 +9,12 @@
 
 
 ### Bug fixes
-- Added test dependencies to `setup.py` (@power-edge)
 
 ### Documentation
-- Added README to tests
 
 ### Other
-
+- Added test dependencies to `setup.py` (@power-edge)
+- Simplify dependencies of `requirements.txt` (remove duplicates from `setup.py`)
 
 
 ## Release 2.2.1 (2023-11-17)

README.md

@@ -52,6 +52,7 @@ Note: you can do `python -m rl_zoo3.train` from any folder and you have access t
 ```
 apt-get install swig cmake ffmpeg
 pip install -r requirements.txt
+pip install -e .[plots,tests]
 ```
 
 Please see [Stable Baselines3 documentation](https://stable-baselines3.readthedocs.io/en/master/) for alternatives to install stable baselines3.

docs/guide/install.rst

@@ -48,6 +48,7 @@ With extra envs and test dependencies:
 
 	apt-get install swig cmake ffmpeg
 	pip install -r requirements.txt
+  pip install -e .[plots,tests]
 
 
 Please see `Stable Baselines3 documentation <https://stable-baselines3.readthedocs.io/en/master/>`_ for alternatives to install stable baselines3.

requirements.txt

@@ -1,22 +1,12 @@
 gym==0.26.2
 stable-baselines3[extra_no_roms,tests,docs]>=2.3.0a1,<3.0
-sb3-contrib>=2.3.0a1,<3.0
 box2d-py==2.3.8
-pybullet
 pybullet_envs_gymnasium>=0.4.0
 # minigrid
-# scikit-optimize
-optuna~=3.0
-pyyaml>=5.1
 cloudpickle>=2.2.1
+# optuna plots:
 plotly
 # need to upgrade to gymnasium:
 # panda-gym~=3.0.1
-rliable>=1.0.5
 wandb
-huggingface_sb3>=3.0,<4.0
-seaborn
-tqdm
-rich
 moviepy
-ruff

setup.py

@@ -26,10 +26,10 @@
 ]
 plots_requires = ["seaborn", "rliable>=1.0.5", "scipy~=1.10"]
 test_requires = [
-    "pytest",
-    "pytablewriter",
-    "gymnasium[mujoco]",
-    "minigrid",
+    # for MuJoCo envs v4:
+    "mujoco~=2.3",
+    # install parking-env to test HER
+    "highway-env==1.8.2",
 ]
 
 setup(
@@ -44,7 +44,7 @@
     },
     entry_points={"console_scripts": ["rl_zoo3=rl_zoo3.cli:main"]},
     install_requires=install_requires,
-    extras_require={"plots": plots_requires, "test": test_requires},
+    extras_require={"plots": plots_requires, "tests": test_requires},
     description="A Training Framework for Stable Baselines3 Reinforcement Learning Agents",
     author="Antonin Raffin",
     url="https://github.com/DLR-RM/rl-baselines3-zoo",