From 1e0226d360df520ec9073ae136ffa3ab39ed2c39 Mon Sep 17 00:00:00 2001 From: Sergei Lebedev Date: Fri, 7 Nov 2025 02:06:51 -0800 Subject: [PATCH] Ignore pytype errors produced with --use-functools-partial-overlay PiperOrigin-RevId: 829338416 Change-Id: I1867bd18a7c57e2faf0f121d89dfc4129c5f31b8 --- mjx/mujoco/mjx/_src/collision_convex.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mjx/mujoco/mjx/_src/collision_convex.py b/mjx/mujoco/mjx/_src/collision_convex.py index 870759ed..0d60c4c9 100644 --- a/mjx/mujoco/mjx/_src/collision_convex.py +++ b/mjx/mujoco/mjx/_src/collision_convex.py @@ -71,7 +71,7 @@ def collider(ncon: int): infos[i] = hfield_info.replace(pos=infos[i].pos, mat=infos[i].mat) in_axes[i] = hfield_info.replace(pos=0, mat=0, data=None) fn = functools.partial(fn, subgrid_size=key.subgrid_size) - dist, pos, frame = jax.vmap(fn, in_axes=in_axes)(*infos) + dist, pos, frame = jax.vmap(fn, in_axes=in_axes)(*infos) # pytype: disable=wrong-keyword-args if ncon > 1: return jax.tree_util.tree_map(jp.concatenate, (dist, pos, frame)) return dist, pos, frame