Add tendon types to MJX name2id and id2name.

PiperOrigin-RevId: 678424789
Change-Id: I2634be0810f8f7dafec597919c512752d2efcfa6
This commit is contained in:
Erik Frey
2024-09-24 15:45:40 -07:00
committed by Copybara-Service
parent 306e52c30f
commit 2785dbc0da
2 changed files with 4 additions and 0 deletions
+2
View File
@@ -200,6 +200,7 @@ def _getnum(m: Union[Model, mujoco.MjModel], obj: mujoco._enums.mjtObj) -> int:
mujoco.mjtObj.mjOBJ_HFIELD: m.nhfield,
mujoco.mjtObj.mjOBJ_PAIR: m.npair,
mujoco.mjtObj.mjOBJ_EQUALITY: m.neq,
mujoco.mjtObj.mjOBJ_TENDON: m.ntendon,
mujoco.mjtObj.mjOBJ_ACTUATOR: m.nu,
mujoco.mjtObj.mjOBJ_SENSOR: m.nsensor,
mujoco.mjtObj.mjOBJ_NUMERIC: m.nnumeric,
@@ -222,6 +223,7 @@ def _getadr(
mujoco.mjtObj.mjOBJ_HFIELD: m.name_hfieldadr,
mujoco.mjtObj.mjOBJ_PAIR: m.name_pairadr,
mujoco.mjtObj.mjOBJ_EQUALITY: m.name_eqadr,
mujoco.mjtObj.mjOBJ_TENDON: m.name_tendonadr,
mujoco.mjtObj.mjOBJ_ACTUATOR: m.name_actuatoradr,
mujoco.mjtObj.mjOBJ_SENSOR: m.name_sensoradr,
mujoco.mjtObj.mjOBJ_NUMERIC: m.name_numericadr,
+2
View File
@@ -768,6 +768,7 @@ class Model(PyTreeNode):
name_meshadr: mesh name pointers (nmesh,)
name_pairadr: geom pair name pointers (npair,)
name_eqadr: equality constraint name pointers (neq,)
name_tendonadr: tendon name pointers (ntendon,)
name_actuatoradr: actuator name pointers (nu,)
name_sensoradr: sensor name pointers (nsensor,)
name_numericadr: numeric name pointers (nnumeric,)
@@ -1081,6 +1082,7 @@ class Model(PyTreeNode):
name_hfieldadr: np.ndarray
name_pairadr: np.ndarray
name_eqadr: np.ndarray
name_tendonadr: np.ndarray
name_actuatoradr: np.ndarray
name_sensoradr: np.ndarray
name_numericadr: np.ndarray