chenlin/Mujoco_WASM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Ignore pytype errors produced with --use-functools-partial-overlay

Browse Source 
PiperOrigin-RevId: 829338416
Change-Id: I1867bd18a7c57e2faf0f121d89dfc4129c5f31b8
This commit is contained in:
Sergei Lebedev
2025-11-07 02:06:51 -08:00
committed by Copybara-Service
parent 252a0d73df
commit 1e0226d360
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mjx/mujoco/mjx/_src/collision_convex.py
+1 -1
View File
@@ -71,7 +71,7 @@ def collider(ncon: int):
infos[i] = hfield_info.replace(pos=infos[i].pos, mat=infos[i].mat) 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) in_axes[i] = hfield_info.replace(pos=0, mat=0, data=None)
fn = functools.partial(fn, subgrid_size=key.subgrid_size) 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: if ncon > 1:
return jax.tree_util.tree_map(jp.concatenate, (dist, pos, frame)) return jax.tree_util.tree_map(jp.concatenate, (dist, pos, frame))
return dist, pos, frame return dist, pos, frame