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

fixed minimal example shown at mjx.rst

Browse Source
This commit is contained in:
JunYoung Kim
2024-05-21 16:32:54 +09:00
parent f0b633de6a
commit 9e0a437e2a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
doc/mjx.rst
+2 -2
View File
@@ -100,7 +100,7 @@ Neither ``mjx.Model`` nor ``mjx.Data`` are meant to be constructed manually. An
.. code-block:: python .. code-block:: python
model = mujoco.MjModel.from_xml_string("...") model = mujoco.MjModel.from_xml_string("...")
mjx_model = mjx.device_put(model) mjx_model = mjx.put_model(model)
mjx_data = mjx.make_data(model) mjx_data = mjx.make_data(model)
Using ``mjx.make_data`` may be preferable when constructing batched ``mjx.Data`` structures inside of a ``vmap``. Using ``mjx.make_data`` may be preferable when constructing batched ``mjx.Data`` structures inside of a ``vmap``.
@@ -151,7 +151,7 @@ Minimal example
""" """
model = mujoco.MjModel.from_xml_string(XML) model = mujoco.MjModel.from_xml_string(XML)
mjx_model = mjx.device_put(model) mjx_model = mjx.put_model(model)
@jax.vmap @jax.vmap
def batched_step(vel): def batched_step(vel):