check default wrappers exist before stripping (#34)

Farama-Foundation · Jan 30, 2023 · f3021b5 · f3021b5 · vercel · Feb 13, 2023
1 parent d8777f5
commit f3021b5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -21,7 +21,7 @@ repos:
           - --show-source
           - --statistics
   - repo: https://github.com/PyCQA/isort
-    rev: 5.10.1
+    rev: 5.12.0
     hooks:
       - id: isort
         args: ["--profile", "black"]

diff --git a/shimmy/openai_gym_compatibility.py b/shimmy/openai_gym_compatibility.py
@@ -283,10 +283,11 @@ def _strip_default_wrappers(env: gym.Env) -> gym.Env:
     Returns:
         The environment without builtin wrappers
     """
-    default_wrappers = (
-        gym.wrappers.render_collection.RenderCollection,
-        gym.wrappers.human_rendering.HumanRendering,
-    )
+    default_wrappers = ()
+    if hasattr(gym.wrappers, "render_collection"):
+        default_wrappers += (gym.wrappers.render_collection.RenderCollection,)
+    if hasattr(gym.wrappers, "human_rendering"):
+        default_wrappers += (gym.wrappers.human_rendering.HumanRendering,)
     while isinstance(env, default_wrappers):
         env = env.env
     return env