Fix #2237.
PiperOrigin-RevId: 703547316 Change-Id: I279876a299e56a16f8d5227fc1d0f64265477853
This commit is contained in:
committed by
Copybara-Service
parent
15cb1fdfb4
commit
c1d97d4a22
@@ -233,9 +233,11 @@ def make_data(
|
||||
solreffriction=jp.zeros((ncon, mujoco.mjNREF), dtype=float),
|
||||
solimp=jp.zeros((ncon, mujoco.mjNIMP), dtype=float),
|
||||
dim=dim,
|
||||
geom1=jp.full((ncon,), -1, dtype=jp.int32),
|
||||
geom2=jp.full((ncon,), -1, dtype=jp.int32),
|
||||
geom=jp.full((ncon, 2), -1, dtype=jp.int32),
|
||||
# let jax pick contact.geom int precision, for interop with
|
||||
# jax_enable_x64
|
||||
geom1=jp.full((ncon,), -1, dtype=int),
|
||||
geom2=jp.full((ncon,), -1, dtype=int),
|
||||
geom=jp.full((ncon, 2), -1, dtype=int),
|
||||
efc_address=efc_address,
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user