Merge pull request #1681 from lgkimjy:dev/mjxdoc_minimal_example_issue

PiperOrigin-RevId: 635887142 Change-Id: Id3c39cacb5db9a1d2ff3a926a5a532839b05586c
2024-05-21 12:14:56 -07:00
parent 2f19941fcf 9e0a437e2a
commit d4afdd8583
1 changed files with 2 additions and 2 deletions
@@ -100,7 +100,7 @@ Neither ``mjx.Model`` nor ``mjx.Data`` are meant to be constructed manually.  An
 .. code-block:: python

   model = mujoco.MjModel.from_xml_string("...")
-   mjx_model = mjx.device_put(model)
+   mjx_model = mjx.put_model(model)
   mjx_data = mjx.make_data(model)

 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)
-   mjx_model = mjx.device_put(model)
+   mjx_model = mjx.put_model(model)

   @jax.vmap
   def batched_step(vel):