From 848c33e16b95a145ad67162ca77705d613095a3a Mon Sep 17 00:00:00 2001
From: peterdavidfagan <peterdavidfagan@gmail.com>
Date: Fri, 17 May 2024 12:22:58 +0100
Subject: [PATCH] fix put_data function signature

---
 doc/mjx.rst | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/doc/mjx.rst b/doc/mjx.rst
index a2e79d8e..22a13490 100644
--- a/doc/mjx.rst
+++ b/doc/mjx.rst
@@ -75,7 +75,7 @@ an ``mjx.Data``:
    model = mujoco.MjModel.from_xml_string("...")
    data = mujoco.MjData(model)
    mjx_model = mjx.put_model(model)
-   mjx_data = mjx.put_data(data)
+   mjx_data = mjx.put_data(model, data)
 
 These MJX variants mirror their MuJoCo counterparts but have a few key differences: