Fix mujoco_py only installation

Only import `mujoco_rendering` when `mujoco` installed. Otherwise the `try import` code in `gym.envs.mujoco.mujoco_env` is useless.
openai · Sep 7, 2022 · 7afacac · 7afacac
1 parent 824507b
commit 7afacac
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/gym/envs/mujoco/__init__.py b/gym/envs/mujoco/__init__.py
@@ -1,8 +1,10 @@
-from gym.envs.mujoco.mujoco_env import MujocoEnv, MuJocoPyEnv  # isort:skip
-from gym.envs.mujoco.mujoco_rendering import (  # isort:skip
-    RenderContextOffscreen,
-    Viewer,
-)
+from gym.envs.mujoco.mujoco_env import MujocoEnv, MuJocoPyEnv, MUJOCO_NOT_INSTALLED  # isort:skip
+if not MUJOCO_NOT_INSTALLED:
+    from gym.envs.mujoco.mujoco_rendering import (  # isort:skip
+        RenderContextOffscreen,
+        Viewer,
+    )
+
 
 # ^^^^^ so that user gets the correct error
 # message if mujoco is not installed correctly