Update env.model.nbvh for mujoco>=3.1.4 (#1012)

Farama-Foundation · Apr 11, 2024 · 94a7909 · 94a7909
1 parent a152867
commit 94a7909
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/envs/mujoco/test_mujoco_v5.py b/tests/envs/mujoco/test_mujoco_v5.py
@@ -593,8 +593,12 @@ def test_model_object_count(version: str):
         assert env.model.nv == 11
         assert env.model.nu == 7
         assert env.model.nbody == 13
-        if mujoco.__version__ >= "3.1.2":
+        if mujoco.__version__ >= "3.1.4":
+            assert env.model.nbvh == 7
+        elif mujoco.__version__ >= "3.1.2":
             assert env.model.nbvh == 8
+        else:
+            assert env.model.nbvh == 18
         assert env.model.njnt == 11
         assert env.model.ngeom == 21
         assert env.model.ntendon == 0