From 5381b5dfe53cd6ebe8fdae4bb8f15295d76e9bd0 Mon Sep 17 00:00:00 2001
From: Kingsley Nweye <etonwana@yahoo.com>
Date: Sat, 3 Feb 2024 20:36:31 -0600
Subject: [PATCH] bug fixes

---
 examples/citylearn_rlem23_tutorial.ipynb |  2 +-
 examples/tutorial.ipynb                  | 13 ++-----------
 2 files changed, 3 insertions(+), 12 deletions(-)

diff --git a/examples/citylearn_rlem23_tutorial.ipynb b/examples/citylearn_rlem23_tutorial.ipynb
index 195347948..70ca4544d 100644
--- a/examples/citylearn_rlem23_tutorial.ipynb
+++ b/examples/citylearn_rlem23_tutorial.ipynb
@@ -2458,7 +2458,7 @@
         "fig, axs = plt.subplots(1, 3, figsize=figsize, sharey=True)\n",
         "axs[0] = plot_table(\n",
         "    axs[0], tql_model.q[0], 'Q-Table',\n",
-        "    'Q-Value', 'State (Hour)', 'Action Index'\n",
+        "    cmap, 'Q-Value', 'State (Hour)', 'Action Index'\n",
         ")\n",
         "axs[1] = plot_table(\n",
         "    axs[1], tql_model.q_exploration[0], 'Q-Table Exploration',\n",
diff --git a/examples/tutorial.ipynb b/examples/tutorial.ipynb
index 06a40ccdb..75445ed64 100644
--- a/examples/tutorial.ipynb
+++ b/examples/tutorial.ipynb
@@ -1,14 +1,5 @@
 {
   "cells": [
-    {
-      "cell_type": "markdown",
-      "metadata": {
-        "id": "view-in-github"
-      },
-      "source": [
-        "<a href=\"https://colab.research.google.com/github/intelligent-environments-lab/CityLearn/blob/master/examples/tutorial.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
-      ]
-    },
     {
       "cell_type": "markdown",
       "metadata": {
@@ -2459,7 +2450,7 @@
         "fig, axs = plt.subplots(1, 3, figsize=figsize, sharey=True)\n",
         "axs[0] = plot_table(\n",
         "    axs[0], tql_model.q[0], 'Q-Table',\n",
-        "    'Q-Value', 'State (Hour)', 'Action Index'\n",
+        "    cmap, 'Q-Value', 'State (Hour)', 'Action Index'\n",
         ")\n",
         "axs[1] = plot_table(\n",
         "    axs[1], tql_model.q_exploration[0], 'Q-Table Exploration',\n",
@@ -3774,4 +3765,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
+}
\ No newline at end of file