chenlin/Mujoco_WASM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix small bug in testspeed.

Browse Source 
PiperOrigin-RevId: 627773755
Change-Id: Iebbcaa41cfdde126312e388d368843a4754abfb8
This commit is contained in:
Baruch Tabanpour
2024-04-24 10:29:39 -07:00
committed by Copybara-Service
parent 1c8ff66cc9
commit 0a60d555c6
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mjx/mujoco/mjx/testspeed.py
+3 -2
View File
@@ -47,8 +47,9 @@ _OUTPUT = flags.DEFINE_enum(
def _main(argv: Sequence[str]): def _main(argv: Sequence[str]):
"""Runs testpeed function.""" """Runs testpeed function."""
base_path = _BASE_PATH.value or epath.resource_path('mujoco.mjx') path = epath.resource_path('mujoco.mjx') / 'test_data'
f = base_path / 'test_data' / _MJCF.value path = _BASE_PATH.value or path
f = epath.Path(path) / _MJCF.value
m = mujoco.MjModel.from_xml_path(f.as_posix()) m = mujoco.MjModel.from_xml_path(f.as_posix())
print(f'Rolling out {_NSTEP.value} steps at dt = {m.opt.timestep:.3f}...') print(f'Rolling out {_NSTEP.value} steps at dt = {m.opt.timestep:.3f}...')