From baef4733c18434db1edee38a73e9374af8959601 Mon Sep 17 00:00:00 2001
From: Thomas Simonini <simonini.thomas.pro@gmail.com>
Date: Sun, 29 May 2022 15:27:59 +0200
Subject: [PATCH] Revert "Fix bug with agent evaluation for very long episodes"

This reverts commit 591c8ccb75e71059bb35d3347469a94590f1d248, reversing
changes made to f74adfb0a27dca47a621e71eda1fa5bb59aa1cb7.
---
 unit2/unit2.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/unit2/unit2.ipynb b/unit2/unit2.ipynb
index 20fbb1c..c94dbf9 100644
--- a/unit2/unit2.ipynb
+++ b/unit2/unit2.ipynb
@@ -901,9 +901,9 @@
         "      total_rewards_ep += reward\n",
         "        \n",
         "      if done:\n",
+        "        episode_rewards.append(total_rewards_ep)\n",
         "        break\n",
         "      state = new_state\n",
-        "    episode_rewards.append(total_rewards_ep)\n",
         "  mean_reward = np.mean(episode_rewards)\n",
         "  std_reward = np.std(episode_rewards)\n",
         "\n",