PiperOrigin-RevId: 703547316
Change-Id: I279876a299e56a16f8d5227fc1d0f64265477853
This commit is contained in:
Baruch Tabanpour
2024-12-06 10:45:37 -08:00
committed by Copybara-Service
parent 15cb1fdfb4
commit c1d97d4a22
+5 -3
View File
@@ -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,
)