From 6bc0bbf7df3058ebfe384022c880ff96239ea1d3 Mon Sep 17 00:00:00 2001 From: Baruch Tabanpour Date: Tue, 13 May 2025 10:51:49 -0700 Subject: [PATCH] internal change PiperOrigin-RevId: 758287280 Change-Id: I8b99a5d44910d5b2e7857b808b1696f244778a26 --- mjx/mujoco/mjx/_src/collision_driver.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mjx/mujoco/mjx/_src/collision_driver.py b/mjx/mujoco/mjx/_src/collision_driver.py index e618fe19..04a4a3d6 100644 --- a/mjx/mujoco/mjx/_src/collision_driver.py +++ b/mjx/mujoco/mjx/_src/collision_driver.py @@ -429,4 +429,4 @@ def collision(m: Model, d: Data) -> Data: contacts = sum([condim_groups[k] for k in sorted(condim_groups)], []) contact = jax.tree_util.tree_map(lambda *x: jp.concatenate(x), *contacts) - return d.replace(_impl=d._impl.replace(contact=contact)) # pytype: disable=attribute-error + return d.tree_replace({'_impl.contact': contact})