Code review fixes DLR-RM#1

As suggested by Antonin Raffin <[email protected]>.

tests/test_env_checker.py

@@ -114,14 +114,7 @@ def step(self, action):
         check_env(env=test_env)
 
 
-class StepCalledAfterEnvTerminatedException(Exception):
-    pass
-
-
 class LimitedStepsTestEnv(gym.Env):
-    metadata = {"render_modes": ["human"]}
-    render_mode = None
-
     action_space = spaces.Discrete(n=2)
     observation_space = spaces.Discrete(n=2)
 
@@ -145,8 +138,7 @@ def reset(self, *, seed: Optional[int] = None, options: Optional[Dict] = None) -
     def step(self, action: np.ndarray) -> Tuple[int, float, bool, bool, Dict[str, Any]]:
         self._steps_called += 1
 
-        if self._terminated:
-            raise StepCalledAfterEnvTerminatedException
+        assert not self._terminated
 
         observation = 0
         reward = 0.0
@@ -155,10 +147,7 @@ def step(self, action: np.ndarray) -> Tuple[int, float, bool, bool, Dict[str, An
 
         return observation, reward, self._terminated, truncated, {}
 
-    def render(self, mode: str = "human") -> None:
-        pass
-
-    def close(self):
+    def render(self) -> None:
         pass