Replace pybullet envs with gymnasium compatible versions.

DLR-RM · Sep 14, 2023 · 186b4a8 · 186b4a8
1 parent eab86ac
commit 186b4a8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -49,6 +49,8 @@ jobs:
         # install parking-env to test HER
         pip install highway-env==1.8.1
         pip install -e .
+        # install pybullet envs
+        pip install pybullet_envs_gymnasium
     - name: Lint with ruff
       run: |
         make lint

diff --git a/rl_zoo3/import_envs.py b/rl_zoo3/import_envs.py
@@ -6,9 +6,9 @@
 from rl_zoo3.wrappers import MaskVelocityWrapper
 
 try:
-    import pybullet_envs  # pytype: disable=import-error
+    import pybullet_envs_gymnasium  # pytype: disable=import-error
 except ImportError:
-    pybullet_envs = None
+    pybullet_envs_gymnasium = None
 
 try:
     import highway_env  # pytype: disable=import-error