Fix small bug in testspeed.

PiperOrigin-RevId: 627773755 Change-Id: Iebbcaa41cfdde126312e388d368843a4754abfb8
2024-04-24 10:29:39 -07:00
parent 1c8ff66cc9
commit 0a60d555c6
1 changed files with 3 additions and 2 deletions
@@ -47,8 +47,9 @@ _OUTPUT = flags.DEFINE_enum(

 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}...')