bug fixes

intelligent-environments-lab · Feb 4, 2024 · 5381b5d · 5381b5d
1 parent fc637b9
commit 5381b5d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 12 deletions.
diff --git 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
@@ -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
-}
+}