update envs requirements

Author: jakegrigsby

.github/workflows/gh-pages.yml

@@ -27,6 +27,7 @@ jobs:
           pip install --upgrade pip
           pip install -e .
           pip install -e .[envs]
+          pip install gymnasium[mujoco]
 
       - name: Install docs dependencies
         run: |

setup.py

@@ -34,6 +34,10 @@
             "dm_alchemy @ git+https://github.com/deepmind/dm_alchemy.git",
             "xminigrid",
             "gymnax",
+            "matplotlib",
+            "opencv-python",
+            "procgen",
+            "minigrid",
         ],
         "flash": ["ninja", "packaging", "flash-attn"],
         "mamba": ["causal-conv1d>=1.1.0", "mamba-ssm"],