From b82c382c53d866b391fde85914ee172634270ef5 Mon Sep 17 00:00:00 2001 From: Tyler Lindberg Date: Tue, 12 Dec 2023 16:28:36 -0800 Subject: [PATCH] Use positional arguments with jp.where --- mjx/tutorial.ipynb | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/mjx/tutorial.ipynb b/mjx/tutorial.ipynb index 522b0208..f27461c6 100644 --- a/mjx/tutorial.ipynb +++ b/mjx/tutorial.ipynb @@ -416,10 +416,8 @@ " forward_reward = self._forward_reward_weight * velocity[0]\n", "\n", " min_z, max_z = self._healthy_z_range\n", - " is_healthy = jp.where(data.qpos[2] \u003c min_z, x=0.0, y=1.0)\n", - " is_healthy = jp.where(\n", - " data.qpos[2] \u003e max_z, x=0.0, y=is_healthy\n", - " )\n", + " is_healthy = jp.where(data.qpos[2] \u003c min_z, 0.0, 1.0)\n", + " is_healthy = jp.where(data.qpos[2] \u003e max_z, 0.0, is_healthy)\n", " if self._terminate_when_unhealthy:\n", " healthy_reward = self._healthy_reward\n", " else:\n",