Merge pull request #125 from discordianfish/patch-1

unit2: Use tqdm in train template

notebooks/unit2/requirements-unit2.txt

@@ -8,3 +8,4 @@ pyyaml==6.0
 imageio
 imageio_ffmpeg
 pyglet==1.5.1
+tqdm

notebooks/unit2/unit2.ipynb

@@ -280,6 +280,7 @@
         "import random\n",
         "import imageio\n",
         "import os\n",
+        "import tqdm\n",
         "\n",
         "import pickle5 as pickle\n",
         "from tqdm.notebook import tqdm"
@@ -771,7 +772,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",
@@ -1747,4 +1748,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
+}