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

Set some visual params as jax.Arrays for visual domain rando.

Browse Source 
PiperOrigin-RevId: 692273879
Change-Id: I80475aa65744de06a6bfebbfb95a44faa751595f
This commit is contained in:
Baruch Tabanpour
2024-11-01 13:23:22 -07:00
committed by Copybara-Service
parent b6037d1759
commit 47ebb0614e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mjx/mujoco/mjx/_src/types.py
+2 -2
View File
@@ -963,8 +963,8 @@ class Model(PyTreeNode):
light_bodyid: np.ndarray = _restricted_to('mujoco')
light_targetbodyid: np.ndarray = _restricted_to('mujoco')
light_directional: np.ndarray
light_pos: np.ndarray = _restricted_to('mujoco')
light_dir: np.ndarray = _restricted_to('mujoco')
light_pos: jax.Array
light_dir: jax.Array
light_poscom0: np.ndarray = _restricted_to('mujoco')
light_pos0: np.ndarray
light_dir0: np.ndarray