Update test_action_dim_check.py

Farama-Foundation · Jun 28, 2024 · fd7b1e7 · fd7b1e7
1 parent 3d6461a
commit fd7b1e7
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/envs/test_action_dim_check.py b/tests/envs/test_action_dim_check.py
@@ -135,9 +135,7 @@ def test_box_actions_out_of_bound(env: gym.Env):
             assert np.all(obs == oob_obs)
 
         if is_lower_bound:
-            obs, _, _, _, _ = env.step(
-                lower_bounds
-            )  # `env` is unwrapped, and in new step API
+            obs, _, _, _, _ = env.step(lower_bounds)
             oob_action = lower_bounds.copy()
             oob_action[i] -= np.asarray(OOB_VALUE, dtype=dtype)