Fix MJX contact sensor.

PiperOrigin-RevId: 791317188 Change-Id: I97e2020f99742d9bb3cb798d4b5354b51fdda2e2
2025-08-05 12:11:05 -07:00
parent 784c7a419b
commit 3821ff1dcf
1 changed files with 1 additions and 1 deletions
@@ -684,7 +684,7 @@ def sensor_acc(m: Model, d: Data) -> Data:
          slot.append(flip[:, 2, None] * d._impl.contact.frame[cid, 1])

        found = is_contact[cid] & match
-        sensors.append((found[:, None] * np.hstack(slot)).reshape(-1))
+        sensors.append((found[:, None] * jp.hstack(slot)).reshape(-1))
        adrs.append(
            (adr[idx_ds][:, None] + np.arange(num * size)[None]).reshape(-1)
        )