From 93eb47d54ac918c12d9f01a66d2feafddc5d62f2 Mon Sep 17 00:00:00 2001 From: Christof Kaufmann <christof.kaufmann@hs-bochum.de> Date: Fri, 7 Mar 2025 14:34:57 +0100 Subject: [PATCH] Update gymnasium code to v1.0.0 --- gym-graphic-output.ipynb | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/gym-graphic-output.ipynb b/gym-graphic-output.ipynb index c051430..745c16a 100644 --- a/gym-graphic-output.ipynb +++ b/gym-graphic-output.ipynb @@ -16,7 +16,7 @@ }, { "cell_type": "code", - "execution_count": 8, + "execution_count": null, "id": "b425cecc-4212-434f-b57c-a9292c5a08d1", "metadata": { "tags": [] @@ -24,11 +24,12 @@ "outputs": [], "source": [ "import gymnasium as gym\n", + "import ale_py # noqa # pylint: disable=unused-import\n", "\n", - "# env = gym.make(\"Pong-v4\", render_mode=\"rgb_array\")\n", + "# env = gym.make(\"ALE/Pong-v5\", render_mode=\"rgb_array\")\n", "# env.metadata['render_fps'] = 30\n", "\n", - "env = gym.make(\"LunarLander-v2\", render_mode=\"rgb_array\")\n", + "env = gym.make(\"LunarLander-v3\", render_mode=\"rgb_array\")\n", "\n", "env = gym.wrappers.RecordVideo(env, './video/', disable_logger=True,\n", " episode_trigger=lambda episode_id: episode_id % 5 == 0) # every 5th episode\n", @@ -61,23 +62,23 @@ "outputs": [], "source": [ "import gymnasium as gym\n", + "import ale_py # noqa # pylint: disable=unused-import\n", "import PIL.Image\n", "import os\n", "\n", - "# env = gym.make(\"Pong-v4\", render_mode=\"rgb_array\")\n", + "# env = gym.make(\"ALE/Pong-v5\", render_mode=\"rgb_array_list\")\n", "# env.metadata['render_fps'] = 30\n", "\n", "env = gym.make(\"CartPole-v1\", render_mode=\"rgb_array_list\")\n", "\n", "def save_gif(frames, filename='gym.gif'):\n", " directory = os.path.dirname(filename)\n", - " if not os.path.exists(directory):\n", - " os.makedirs(directory)\n", + " os.makedirs(directory, exist_ok=True)\n", "\n", " images = [PIL.Image.fromarray(frame) for frame in frames]\n", " images[0].save(filename, format='GIF', append_images=images[1:],\n", " save_all=True, duration=30, loop=0)\n", - " \n", + "\n", "observation, info = env.reset()\n", "gif_count = 0\n", "for _ in range(1000):\n", @@ -111,11 +112,12 @@ "outputs": [], "source": [ "import gymnasium as gym\n", + "import ale_py # noqa # pylint: disable=unused-import\n", "import matplotlib.pyplot as plt\n", "# you maybe have to restart your kernel, if you changed the matplotlib backend before\n", "%matplotlib widget\n", "\n", - "env = gym.make(\"LunarLander-v2\", render_mode=\"rgb_array\")\n", + "env = gym.make(\"LunarLander-v3\", render_mode=\"rgb_array\")\n", "observation, info = env.reset() # is required once before calling env.render()\n", "img = plt.imshow(env.render()) # only call this once to bring up the plot\n", "plt.show()\n", -- GitLab