From 9c89fe4143e947ef341beb5871394a021db13e94 Mon Sep 17 00:00:00 2001
From: Charlie Crane <charlie.crane@t-s.qinetiq.com>
Date: Tue, 30 Apr 2024 16:14:34 +0100
Subject: [PATCH] #2523 - Typo corrections within notebooks that stopped them
 from running

---
 .../notebooks/Data-Manipulation-E2E-Demonstration.ipynb         | 2 --
 src/primaite/notebooks/Training-an-SB3-Agent.ipynb              | 2 +-
 2 files changed, 1 insertion(+), 3 deletions(-)

diff --git a/src/primaite/notebooks/Data-Manipulation-E2E-Demonstration.ipynb b/src/primaite/notebooks/Data-Manipulation-E2E-Demonstration.ipynb
index 201bf518..9315f782 100644
--- a/src/primaite/notebooks/Data-Manipulation-E2E-Demonstration.ipynb
+++ b/src/primaite/notebooks/Data-Manipulation-E2E-Demonstration.ipynb
@@ -476,9 +476,7 @@
    "outputs": [],
    "source": [
     "obs, reward, terminated, truncated, info = env.step(9)  # scan database file\n",
-    "print(info)\n",
     "obs, reward, terminated, truncated, info = env.step(1)  # scan webapp service\n",
-    "print(info)\n",
     "pprint(obs['NODES'])"
    ]
   },
diff --git a/src/primaite/notebooks/Training-an-SB3-Agent.ipynb b/src/primaite/notebooks/Training-an-SB3-Agent.ipynb
index 140df1b8..ee51aa58 100644
--- a/src/primaite/notebooks/Training-an-SB3-Agent.ipynb
+++ b/src/primaite/notebooks/Training-an-SB3-Agent.ipynb
@@ -59,7 +59,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "gym = PrimaiteGymEnv(game_config=cfg)"
+    "gym = PrimaiteGymEnv(env_config=cfg)"
    ]
   },
   {