Backport 3.3.1 fixes into Core

src/primaite/notebooks/Action-masking.ipynb

@@ -11,6 +11,15 @@
     "PrimAITE environments support action masking. The action mask shows which of the agent's actions are applicable with the current environment state. For example, a node can only be turned on if it is currently turned off."
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!primaite setup"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,

src/primaite/notebooks/Data-Manipulation-Customising-Red-Agent.ipynb

@@ -15,6 +15,15 @@
     "*(For a full explanation of the Data Manipulation scenario, check out the data manipulation scenario notebook)*"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!primaite setup"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,

src/primaite/notebooks/Data-Manipulation-E2E-Demonstration.ipynb

@@ -371,6 +371,15 @@
     "First, load the required modules"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!primaite setup"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,

src/primaite/notebooks/Getting-Information-Out-Of-PrimAITE.ipynb

@@ -9,6 +9,15 @@
     "© Crown-owned copyright 2024, Defence Science and Technology Laboratory UK\n"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!primaite setup"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,

src/primaite/notebooks/Requests-and-Responses.ipynb

@@ -25,6 +25,15 @@
     "Let's set up a minimal network simulation and send some requests to see how it works."
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!primaite setup"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,

src/primaite/notebooks/Terminal-Processing.ipynb

@@ -18,6 +18,15 @@
     "The Terminal service comes pre-installed on most Nodes (The exception being Switches, as these are currently dumb). "
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!primaite setup"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,

src/primaite/notebooks/Training-an-RLLIB-MARL-System.ipynb

@@ -18,6 +18,15 @@
     "#### First, Import packages and read our config file."
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!primaite setup"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -32,8 +41,6 @@
     "from ray.rllib.algorithms.ppo import PPOConfig\n",
     "from primaite.session.ray_envs import PrimaiteRayMARLEnv\n",
     "\n",
-    "# If you get an error saying this config file doesn't exist, you may need to run `primaite setup` in your command line\n",
-    "# to copy the files to your user data path.\n",
     "with open(PRIMAITE_PATHS.user_config_path / 'example_config/data_manipulation_marl.yaml', 'r') as f:\n",
     "    cfg = yaml.safe_load(f)\n",
     "\n",

src/primaite/notebooks/Training-an-RLLib-Agent.ipynb

@@ -11,6 +11,15 @@
     "This notebook will demonstrate how to use PrimaiteRayEnv to train a basic PPO agent."
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!primaite setup"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,

src/primaite/notebooks/Training-an-SB3-Agent.ipynb

@@ -18,6 +18,15 @@
     "#### First, we import the inital packages and read in our configuration file."
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!primaite setup"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,

src/primaite/notebooks/multi-processing.ipynb

@@ -18,6 +18,15 @@
     "Import packages and read config file."
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!primaite setup"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,