diff --git a/tests/envs/mujoco/test_mujoco_rendering.py b/tests/envs/mujoco/test_mujoco_rendering.py index d79b40ce3..750fd1196 100644 --- a/tests/envs/mujoco/test_mujoco_rendering.py +++ b/tests/envs/mujoco/test_mujoco_rendering.py @@ -94,7 +94,7 @@ def test_camera_id(): env_b.reset(seed=5) env_c.reset(seed=5) - assert env_a.render() == env_b.render(), "If this fails, the test is not valid" + assert (env_a.render() == env_b.render()).all(), "If this fails, the test is not valid" assert ( - env_a.render() != env_c.render() + (env_a.render() != env_c.render()).all() ), "render() output should be different for different camera_id"