From fd96de330e68a0690da75ecf8a0709ccf599e28a Mon Sep 17 00:00:00 2001 From: Baruch Tabanpour Date: Tue, 19 Aug 2025 15:11:32 -0700 Subject: [PATCH] Updates to Playground for warp. PiperOrigin-RevId: 797037846 Change-Id: I3828b48aead8a7534a4e8cd735c96017353daa3b --- mjx/mujoco/mjx/_src/io.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mjx/mujoco/mjx/_src/io.py b/mjx/mujoco/mjx/_src/io.py index 165b5066..e3a3936e 100644 --- a/mjx/mujoco/mjx/_src/io.py +++ b/mjx/mujoco/mjx/_src/io.py @@ -1301,7 +1301,7 @@ def _get_data_into_warp( ) result_i = result[i] if batched else result ncon = d_i._impl.ncon[0] - nefc = int(d_i._impl.nefc[0]) + nefc = int(d_i._impl.nefc) # nj = int(d_i._impl.nj[0]) nj = 0 # TODO(btaba): add nj back @@ -1322,7 +1322,7 @@ def _get_data_into_warp( value = getattr(d_i, field.name) if field.name in ('ne', 'nl', 'nf'): - value = value[0] + pass elif field.name in ('nefc', 'ncon'): value = {'nefc': nefc, 'ncon': ncon}[field.name] elif field.name.endswith('xmat') or field.name == 'ximat':