Add width and height arguments to mujoco rendering max geoms test

Farama-Foundation · Oct 5, 2023 · f8ae8f4 · f8ae8f4
1 parent b5d8bc6
commit f8ae8f4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/envs/mujoco/test_mujoco_rendering.py b/tests/envs/mujoco/test_mujoco_rendering.py
@@ -63,7 +63,9 @@ def test_max_geom_attribute(
     """Test that the max_geom attribute is set correctly."""
 
     # initialize renderer
-    renderer = ExposedViewerRenderer(model, data, max_geom=max_geom)
+    renderer = ExposedViewerRenderer(
+        model, data, width=DEFAULT_SIZE, height=DEFAULT_SIZE, max_geom=max_geom
+    )
 
     # assert max_geom attribute
     assert renderer.max_geom == max_geom