Address PR feedback.

erikfrey · erikfrey · commit 639902ea1c3b · 2025-03-17T00:00:04.000-07:00
diff --git a/mujoco_warp/testspeed.py b/mujoco_warp/testspeed.py
@@ -97,6 +97,7 @@ def _main(argv: Sequence[str]):
   print(
     f"Model nbody: {m.nbody} nv: {m.nv} ngeom: {m.ngeom} is_sparse: {_IS_SPARSE.value} solver: {_SOLVER.value}"
   )
+  print(f"Params nconmax: {_NCONMAX.value} njmax: {_NJMAX.value}")
   print(f"Data ncon: {d.ncon} nefc: {d.nefc} keyframe: {_KEYFRAME.value}")
   print(f"Rolling out {_NSTEP.value} steps at dt = {m.opt.timestep:.3f}...")
   jit_time, run_time, trace, steps, ncon, nefc = mjwarp.benchmark(

Original file line number	Diff line number	Diff line change
`@@ -97,6 +97,7 @@ def _main(argv: Sequence[str]):`
`97`	`97`	`print(`
`98`	`98`	`f"Model nbody: {m.nbody} nv: {m.nv} ngeom: {m.ngeom} is_sparse: {_IS_SPARSE.value} solver: {_SOLVER.value}"`
`99`	`99`	`)`
	`100`	`+ print(f"Params nconmax: {_NCONMAX.value} njmax: {_NJMAX.value}")`
`100`	`101`	`print(f"Data ncon: {d.ncon} nefc: {d.nefc} keyframe: {_KEYFRAME.value}")`
`101`	`102`	`print(f"Rolling out {_NSTEP.value} steps at dt = {m.opt.timestep:.3f}...")`
`102`	`103`	`jit_time, run_time, trace, steps, ncon, nefc = mjwarp.benchmark(`