From 4ca56568ca4a90cc36c668476b6bbb0a81ae3df4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Aur=C3=A9lien=20Geron?= Date: Sun, 25 Sep 2022 22:05:11 +1300 Subject: [PATCH] Fix render code for LunarLander-v2 --- 18_reinforcement_learning.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/18_reinforcement_learning.ipynb b/18_reinforcement_learning.ipynb index ba2379a..a2b20a5 100644 --- a/18_reinforcement_learning.ipynb +++ b/18_reinforcement_learning.ipynb @@ -2547,12 +2547,12 @@ "source": [ "def lander_render_policy_net(model, n_max_steps=500, seed=42):\n", " frames = []\n", - " env = gym.make(\"LunarLander-v2\")\n", + " env = gym.make(\"LunarLander-v2\", render_mode=\"rgb_array\")\n", " tf.random.set_seed(seed)\n", " np.random.seed(seed)\n", " obs, info = env.reset(seed=seed)\n", " for step in range(n_max_steps):\n", - " frames.append(env.render(mode=\"rgb_array\"))\n", + " frames.append(env.render())\n", " probas = model(obs[np.newaxis])\n", " logits = tf.math.log(probas + tf.keras.backend.epsilon())\n", " action = tf.random.categorical(logits, num_samples=1)\n",