Improve Wrapper and VectorWrapper error checking (#1260)

Farama-Foundation · Nov 22, 2024 · 13230f4 · 13230f4
1 parent 9ba3283
commit 13230f4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/gymnasium/core.py b/gymnasium/core.py
@@ -307,7 +307,9 @@ def __init__(self, env: Env[ObsType, ActType]):
             env: The environment to wrap
         """
         self.env = env
-        assert isinstance(env, Env)
+        assert isinstance(
+            env, Env
+        ), f"Expected env to be a `gymnasium.Env` but got {type(env)}"
 
         self._action_space: spaces.Space[WrapperActType] | None = None
         self._observation_space: spaces.Space[WrapperObsType] | None = None

diff --git a/gymnasium/vector/vector_env.py b/gymnasium/vector/vector_env.py
@@ -355,7 +355,9 @@ def __init__(self, env: VectorEnv):
             env: The environment to wrap
         """
         self.env = env
-        assert isinstance(env, VectorEnv)
+        assert isinstance(
+            env, VectorEnv
+        ), f"Expected env to be a `gymnasium.vector.VectorEnv` but got {type(env)}"
 
         self._observation_space: gym.Space | None = None
         self._action_space: gym.Space | None = None