From c3d87e64c0816c1429c00ece64f4bb36652c078f Mon Sep 17 00:00:00 2001
From: Tyler Lindberg <tylindberg@hotmail.com>
Date: Fri, 15 Dec 2023 18:34:15 -0800
Subject: [PATCH] Explicitly convert arrays to scalars

---
 python/tutorial.ipynb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/tutorial.ipynb b/python/tutorial.ipynb
index cdc5056671..bad84bdda5 100644
--- a/python/tutorial.ipynb
+++ b/python/tutorial.ipynb
@@ -1080,7 +1080,7 @@
         "  for i in range(n_steps):\n",
         "    mujoco.mj_step(model, data)\n",
         "    sim_time[i] = data.time\n",
-        "    angle[i] = data.joint('root').qpos\n",
+        "    angle[i] = data.joint('root').qpos[0]\n",
         "    energy[i] = data.energy[0] + data.energy[1]\n",
         "\n",
         "  # plot\n",
@@ -1198,7 +1198,7 @@
         "  for i in range(n_steps):\n",
         "    mujoco.mj_step(model, data)\n",
         "    if data.warning.number.any():\n",
-        "      warning_index = np.nonzero(data.warning.number)[0]\n",
+        "      warning_index = np.nonzero(data.warning.number)[0][0]\n",
         "      warning = mujoco.mjtWarning(warning_index).name\n",
         "      print(f'stopped due to divergence ({warning}) at timestep {i}.\\n')\n",
         "      break\n",