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

Set geom_matid as a jax.Array for visual domain randomization.

Browse Source 
PiperOrigin-RevId: 700706160
Change-Id: I13ef4da8cd3a671446e4bdd186dcacb4d0b3fb00
This commit is contained in:
Baruch Tabanpour
2024-11-27 08:12:35 -08:00
committed by Copybara-Service
parent 203602b01f
commit f2e9e38edf
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/types.py
+1 -1
View File
@@ -933,7 +933,7 @@ class Model(PyTreeNode):
geom_sameframe: np.ndarray
geom_dataid: np.ndarray
geom_group: np.ndarray
geom_matid: np.ndarray
geom_matid: jax.Array
geom_priority: np.ndarray
geom_solmix: jax.Array
geom_solref: jax.Array