From 579f5782c618986218580faba5e4550341a78325 Mon Sep 17 00:00:00 2001
From: Caleb Robinson <calebrob6@gmail.com>
Date: Sun, 27 Feb 2022 20:12:04 +0000
Subject: [PATCH] Fixed bug in trainer notebook (#434)

---
 docs/tutorials/trainers.ipynb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docs/tutorials/trainers.ipynb b/docs/tutorials/trainers.ipynb
index e8cc20b55ec..f37dd40d202 100644
--- a/docs/tutorials/trainers.ipynb
+++ b/docs/tutorials/trainers.ipynb
@@ -492,13 +492,13 @@
     "        csv_reader = csv.DictReader(f, delimiter=',')\n",
     "        for i, row in enumerate(csv_reader):\n",
     "            try:\n",
-    "                train_rmse.append(float(row[\"train_rmse\"]))\n",
+    "                train_rmse.append(float(row[\"train_RMSE\"]))\n",
     "                train_steps.append(i)\n",
     "            except ValueError: # Ignore rows where train RMSE is empty\n",
     "                pass\n",
     "\n",
     "            try:\n",
-    "                val_rmse.append(float(row[\"val_rmse\"]))\n",
+    "                val_rmse.append(float(row[\"val_RMSE\"]))\n",
     "                val_steps.append(i)\n",
     "            except ValueError: # Ignore rows where val RMSE is empty\n",
     "                pass"