Fix small bug in testspeed.

PiperOrigin-RevId: 627773755 Change-Id: Iebbcaa41cfdde126312e388d368843a4754abfb8
google-deepmind · Apr 24, 2024 · 0a60d55 · 0a60d55
1 parent 1c8ff66
commit 0a60d55
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/mjx/mujoco/mjx/testspeed.py b/mjx/mujoco/mjx/testspeed.py
@@ -47,8 +47,9 @@
 
 def _main(argv: Sequence[str]):
   """Runs testpeed function."""
-  base_path = _BASE_PATH.value or epath.resource_path('mujoco.mjx')
-  f = base_path / 'test_data' / _MJCF.value
+  path = epath.resource_path('mujoco.mjx') / 'test_data'
+  path = _BASE_PATH.value or path
+  f = epath.Path(path) / _MJCF.value
   m = mujoco.MjModel.from_xml_path(f.as_posix())
 
   print(f'Rolling out {_NSTEP.value} steps at dt = {m.opt.timestep:.3f}...')