fix

Farama-Foundation · Sep 27, 2024 · 3eedf9a · 3eedf9a
1 parent 28485c5
commit 3eedf9a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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"