From d4657698bec1acdcc64c8ec997f960216d904813 Mon Sep 17 00:00:00 2001
From: Johannes 'fish' Ziemke <github@5pi.de>
Date: Wed, 14 Dec 2022 17:22:42 +0100
Subject: [PATCH] unit2: Use tqdm in train template

---
 notebooks/unit2/unit2.ipynb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/notebooks/unit2/unit2.ipynb b/notebooks/unit2/unit2.ipynb
index ea78857..a10b711 100644
--- a/notebooks/unit2/unit2.ipynb
+++ b/notebooks/unit2/unit2.ipynb
@@ -771,7 +771,7 @@
       "outputs": [],
       "source": [
         "def train(n_training_episodes, min_epsilon, max_epsilon, decay_rate, env, max_steps, Qtable):\n",
-        "  for episode in range(n_training_episodes):\n",
+        "  for episode in tqdm(range(n_training_episodes))\n",
         "    # Reduce epsilon (because we need less and less exploration)\n",
         "    epsilon = min_epsilon + (max_epsilon - min_epsilon)*np.exp(-decay_rate*episode)\n",
         "    # Reset the environment\n",
@@ -1748,4 +1748,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
\ No newline at end of file
+}