diff --git a/mjx/tutorial.ipynb b/mjx/tutorial.ipynb index 7f13bb37..024a9c8f 100644 --- a/mjx/tutorial.ipynb +++ b/mjx/tutorial.ipynb @@ -909,6 +909,9 @@ "source": [ "#@title Barkour vb Quadruped Env\n", "\n", + "BARKOUR_ROOT_PATH = epath.Path('mujoco_menagerie/google_barkour_vb')\n", + "\n", + "\n", "def get_config():\n", " \"\"\"Returns reward config for barkour quadruped environment.\"\"\"\n", "\n", @@ -979,8 +982,7 @@ " scene_file: str = 'scene_mjx.xml',\n", " **kwargs,\n", " ):\n", - "# path = epath.Path('mujoco_menagerie/google_barkour_vb')\n", - " path = path / scene_file\n", + " path = BARKOUR_ROOT_PATH / scene_file\n", " sys = mjcf.load(path.as_posix())\n", " self._dt = 0.02 # this environment is 50 fps\n", " sys = sys.tree_replace({'opt.timestep': 0.004})\n",