From 5f55e96810944030bcd1e1633bbc818f748430bc Mon Sep 17 00:00:00 2001
From: Taylor Howell <taylorhowell@google.com>
Date: Fri, 10 Oct 2025 09:43:27 -0700
Subject: [PATCH] Fix _get_data_into_warp.

PiperOrigin-RevId: 817681130
Change-Id: Iceb622e5655021ac25388ec28f1f78fe61f58ddc
---
 mjx/mujoco/mjx/_src/io.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mjx/mujoco/mjx/_src/io.py b/mjx/mujoco/mjx/_src/io.py
index ea47f2d2..0a169b04 100644
--- a/mjx/mujoco/mjx/_src/io.py
+++ b/mjx/mujoco/mjx/_src/io.py
@@ -1264,7 +1264,7 @@ def _get_data_into_warp(
         else d
     )
     result_i = result[i] if batched else result
-    ncon = d_i._impl.ncon[0]
+    ncon = d_i._impl.nacon[0]
     nefc = int(d_i._impl.nefc)
     # nj = int(d_i._impl.nj[0])
     nj = 0  # TODO(btaba): add nj back