Update test_mujoco_v5.py

Farama-Foundation · Jun 5, 2024 · 0df9473 · 0df9473
1 parent 52c04fb
commit 0df9473
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/envs/mujoco/test_mujoco_v5.py b/tests/envs/mujoco/test_mujoco_v5.py
@@ -600,7 +600,10 @@ def test_model_object_count(version: str):
         else:
             assert env.model.nbvh == 18
         assert env.model.njnt == 11
-        assert env.model.ngeom == 21
+        if version == "v4":
+            assert env.model.ngeom == 21
+        else:
+            assert env.model.ngeom == 20        
         assert env.model.ntendon == 0
 
     env = gym.make(f"Reacher-{version}").unwrapped