From 4f5aa43f92ae20781f49019286a27fcab0120da9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Aur=C3=A9lien=20Geron?= Date: Fri, 19 Jan 2024 19:01:59 +1300 Subject: [PATCH] Remove a warning when computing left_proba --- 18_reinforcement_learning.ipynb | 2 +- environment.yml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/18_reinforcement_learning.ipynb b/18_reinforcement_learning.ipynb index 0966851..9d32862 100644 --- a/18_reinforcement_learning.ipynb +++ b/18_reinforcement_learning.ipynb @@ -807,7 +807,7 @@ "# extra code – a function that creates an animation for a given policy model\n", "\n", "def pg_policy(obs):\n", - " left_proba = model.predict(obs[np.newaxis], verbose=0)\n", + " left_proba = model.predict(obs[np.newaxis], verbose=0)[0][0]\n", " return int(np.random.rand() > left_proba)\n", "\n", "np.random.seed(42)\n", diff --git a/environment.yml b/environment.yml index bfffaa2..aaf25d4 100644 --- a/environment.yml +++ b/environment.yml @@ -40,6 +40,6 @@ dependencies: - tensorflow~=2.14.0 # Deep Learning library - transformers~=4.35.0 # Natural Language Processing lib for TF or PyTorch - urlextract~=1.8.0 # optionally used in chapter 3, exercise 4 - - gym[classic_control,atari,accept-rom-license] # used only in ch18 + - gymnasium[classic_control,atari,accept-rom-license] # used only in ch18 - google-cloud-aiplatform~=1.36.2 # used only in chapter 19 - google-cloud-storage~=2.13.0 # used only in chapter 19