From 8c5154c9a32686ad44140dc653d27456de5ef0c3 Mon Sep 17 00:00:00 2001 From: Baruch Tabanpour Date: Fri, 24 Oct 2025 15:46:24 -0700 Subject: [PATCH] Fix testspeed for mjx-warp. PiperOrigin-RevId: 823691704 Change-Id: If90f6137915de134fdda3c7616bc810555c25d87 --- mjx/mujoco/mjx/warp/testspeed.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mjx/mujoco/mjx/warp/testspeed.py b/mjx/mujoco/mjx/warp/testspeed.py index a6bea516..9aa24b0d 100644 --- a/mjx/mujoco/mjx/warp/testspeed.py +++ b/mjx/mujoco/mjx/warp/testspeed.py @@ -44,7 +44,7 @@ _NSTEP = flags.DEFINE_integer('nstep', 1000, 'number of steps per rollout') _NENV = flags.DEFINE_integer('nenv', 8192, 'number of environments to simulate') _UNROLL = flags.DEFINE_integer('unroll', 4, 'number of steps to unroll') _NCONMAX = flags.DEFINE_integer('nconmax', 30_000, 'max contacts') -_NJMAX = flags.DEFINE_integer('njmax', 80_000, 'max constraints') +_NJMAX = flags.DEFINE_integer('njmax', 10, 'max constraints per world') _WP_KERNEL_CACHE_DIR = flags.DEFINE_string( 'wp_kernel_cache_dir', None,