Replace deprecated function.

PiperOrigin-RevId: 627568947 Change-Id: Id3c7f0b17895b3400f34ecc7bbb78007f37a54ba
google-deepmind · Apr 24, 2024 · cc3868e · cc3868e
1 parent 8d283c9
commit cc3868e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/model_test.py b/test/model_test.py
@@ -89,7 +89,7 @@ class MjxModelsTest(parameterized.TestCase):
   @parameterized.parameters(_MJX_MODEL_XMLS)
   def test_compiles_and_steps(self, xml_path: pathlib.Path) -> None:
     model = mujoco.MjModel.from_xml_path(str(xml_path))
-    model = mjx.device_put(model)
+    model = mjx.put_model(model)
     data = mjx.make_data(model)
     ctrlrange = jp.where(
         model.actuator_ctrllimited[:, None],