From 9fc1ac2c7d049d557a83be149f10d775a3541f98 Mon Sep 17 00:00:00 2001
From: Alessio Quaglino <quaglino@google.com>
Date: Thu, 16 Jan 2025 09:04:09 -0800
Subject: [PATCH] Fix typo in sensordata binding.

PiperOrigin-RevId: 716257343
Change-Id: I4da12d8cda4342eb9f5a4df664bb43841db8e332
---
 mjx/mujoco/mjx/_src/support.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/mjx/mujoco/mjx/_src/support.py b/mjx/mujoco/mjx/_src/support.py
index 905a2cbf..748be165 100644
--- a/mjx/mujoco/mjx/_src/support.py
+++ b/mjx/mujoco/mjx/_src/support.py
@@ -444,8 +444,8 @@ class BindData(object):
       num = self.model.sensor_dim[self.id]
       if isinstance(self.id, list):
         idx = []
-        for i, n in zip(self.id, num):
-          idx.extend(adr[i] + j for j in range(n))
+        for a, n in zip(adr, num):
+          idx.extend(a + j for j in range(n))
         return getattr(self.data, name)[idx, ...]
       else:
         return getattr(self.data, name)[adr : adr + num, ...]