#2871 - Fix notebook failure

2024-09-17 16:19:43 +01:00
parent 3a5b75239d
commit 8d3760b5a7
1 changed files with 5 additions and 0 deletions
--- 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)"