From 8d3760b5a7e8bf53f8a7e20cabc3a5597ecd897f Mon Sep 17 00:00:00 2001
From: Charlie Crane <Charlie.Crane@t-s.qinetiq.com>
Date: Tue, 17 Sep 2024 16:19:43 +0100
Subject: [PATCH] #2871 - Fix notebook failure

---
 .../notebooks/Getting-Information-Out-Of-PrimAITE.ipynb      | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/primaite/notebooks/Getting-Information-Out-Of-PrimAITE.ipynb b/src/primaite/notebooks/Getting-Information-Out-Of-PrimAITE.ipynb
index e4009822..6a60c1bc 100644
--- a/src/primaite/notebooks/Getting-Information-Out-Of-PrimAITE.ipynb
+++ b/src/primaite/notebooks/Getting-Information-Out-Of-PrimAITE.ipynb
@@ -160,6 +160,11 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "with open(data_manipulation_config_path(), 'r') as f:\n",
+    "    cfg = yaml.safe_load(f)\n",
+    "\n",
+    "env = PrimaiteGymEnv(env_config=cfg)\n",
+    "\n",
     "# Run the training session to generate some resultant data.\n",
     "for i in range(100):\n",
     "    env.step(0)"