Change agents list in game object to dictionary

src/primaite/notebooks/training_example_ray_multi_agent.ipynb

@@ -60,7 +60,7 @@
     "        policies={'defender_1','defender_2'}, # These names are the same as the agents defined in the example config.\n",
     "        policy_mapping_fn=lambda agent_id, episode, worker, **kw: agent_id,\n",
     "        )\n",
-    "    .environment(env=PrimaiteRayMARLEnv, env_config={\"cfg\":cfg})#, disable_env_checking=True)\n",
+    "    .environment(env=PrimaiteRayMARLEnv, env_config=cfg)#, disable_env_checking=True)\n",
     "    .rollouts(num_rollout_workers=0)\n",
     "    .training(train_batch_size=128)\n",
     "    )\n"
@@ -88,6 +88,13 @@
     "    param_space=config\n",
     ").fit()"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {

src/primaite/notebooks/training_example_ray_single_agent.ipynb

@@ -54,7 +54,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "env_config = {\"cfg\":cfg}\n",
+    "env_config = cfg\n",
     "\n",
     "config = (\n",
     "    PPOConfig()\n",

src/primaite/notebooks/training_example_sb3.ipynb

@@ -27,9 +27,7 @@
    "outputs": [],
    "source": [
     "with open(example_config_path(), 'r') as f:\n",
-    "    cfg = yaml.safe_load(f)\n",
-    "\n",
-    "game = PrimaiteGame.from_config(cfg)"
+    "    cfg = yaml.safe_load(f)\n"
    ]
   },
   {
@@ -76,6 +74,13 @@
    "source": [
     "model.save(\"deleteme\")"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {