parabellum

ludens.ipynb

@@ -9,7 +9,7 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "2024-07-07 11:05:57.124 python[56985:875363] ApplePersistence=NO\n"
+      "2024-07-07 16:37:48.299 python[66039:1017127] ApplePersistence=NO\n"
      ]
     }
    ],
@@ -47,12 +47,10 @@
    "outputs": [],
    "source": [
     "state_seq = []\n",
-    "for i in range(30):\n",
+    "for i in range(100):\n",
     "    rng, act_rng, step_key = random.split(rng, 3)\n",
     "    act_key = random.split(act_rng, len(env.agents))\n",
-    "    actions = {\n",
-    "        a: 0 * env.action_space(a).sample(k) for a, k in zip(env.agents, act_key)\n",
-    "    }\n",
+    "    actions = {a: env.action_space(a).sample(k) for a, k in zip(env.agents, act_key)}\n",
     "    state_seq.append((step_key, state, actions))\n",
     "    obs, state, reward, done, info = env.step(step_key, state, actions)"
    ]
@@ -66,7 +64,7 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "240it [00:04, 56.24it/s]                       \n"
+      "800it [00:14, 56.05it/s]                        \n"
      ]
     },
     {
@@ -90,7 +88,15 @@
      "output_type": "stream",
      "text": [
       "Moviepy - Done !\n",
-      "Moviepy - video ready output/parabellum.mp4\n"
+      "Moviepy - video ready output/parabellum.mp4\n",
+      "MoviePy - Building file output/parabellum.gif with imageio.\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "                                                              \r"
      ]
     }
    ],

parabellum/vis.py

@@ -130,7 +130,7 @@ def animate_one(self, state_seq, action_seq, save_fname):
         # save the images
         clip = ImageSequenceClip(frames, fps=48)
         clip.write_videofile(save_fname, fps=48)
-        # clip.write_gif(save_fname.replace(".mp4", ".gif"), fps=24)
+        clip.write_gif(save_fname.replace(".mp4", ".gif"), fps=24)
         pygame.quit()
 
     def render_agents(self, screen, state):