diff --git a/mjx/mujoco/mjx/_src/test_util.py b/mjx/mujoco/mjx/_src/test_util.py index 349cab16..a5a3af01 100644 --- a/mjx/mujoco/mjx/_src/test_util.py +++ b/mjx/mujoco/mjx/_src/test_util.py @@ -53,7 +53,7 @@ def benchmark( nstep: int = 1000, batch_size: int = 1024, unroll_steps: int = 1, - solver: str = 'cg', + solver: str = 'newton', iterations: int = 1, ls_iterations: int = 4, ) -> Tuple[float, float, int]: