Merge pull request #4 from huggingface/user/aliberts/2024_04_09_fix_G…

…L_error Fix GL error
huggingface · Apr 9, 2024 · 824b228 · 824b228
2 parents 6c98935 + 7150dea
commit 824b228
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/gym_pusht/envs/pusht.py b/gym_pusht/envs/pusht.py
@@ -1,4 +1,5 @@
 import collections
+import os
 
 import cv2
 import gymnasium as gym
@@ -12,6 +13,10 @@
 
 from .pymunk_override import DrawOptions
 
+RENDER_MODES = ["rgb_array"]
+if os.environ.get("MUJOCO_GL") != "egl":
+    RENDER_MODES.append("human")
+
 
 def pymunk_to_shapely(body, shapes):
     geoms = []
@@ -114,7 +119,7 @@ class PushTEnv(gym.Env):
     * TODO:
     """
 
-    metadata = {"render_modes": ["human", "rgb_array"], "render_fps": 10}
+    metadata = {"render_modes": RENDER_MODES, "render_fps": 10}
 
     def __init__(
         self,