From 28485c51dc3cda15fc861bfbcb68e5205305169a Mon Sep 17 00:00:00 2001 From: Kallinteris Andreas Date: Fri, 27 Sep 2024 11:52:17 +0300 Subject: [PATCH] fix test --- tests/envs/mujoco/test_mujoco_rendering.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/envs/mujoco/test_mujoco_rendering.py b/tests/envs/mujoco/test_mujoco_rendering.py index cf07fa0d8..d79b40ce3 100644 --- a/tests/envs/mujoco/test_mujoco_rendering.py +++ b/tests/envs/mujoco/test_mujoco_rendering.py @@ -86,9 +86,9 @@ def test_max_geom_attribute( def test_camera_id(): """Assert that the camera_id parameter works correctly.""" - env_a = gymnasium.make("Ant-v5", camera_id=0) - env_b = gymnasium.make("Ant-v5", camera_id=0) - env_c = gymnasium.make("Ant-v5", camera_id=1) + env_a = gymnasium.make("Ant-v5", camera_id=0, render_mode="rgb_array").unwrapped + env_b = gymnasium.make("Ant-v5", camera_id=0, render_mode="rgb_array").unwrapped + env_c = gymnasium.make("Ant-v5", camera_id=1, render_mode="rgb_array").unwrapped env_a.reset(seed=5) env_b.reset(seed=5)