Enh: Save metadata every round

Closes #58

Author: klieret

codeclash/tournaments/pvp.py

@@ -80,6 +80,8 @@ def run_competition_phase(self, round_num: int) -> None:
         results_file = self.game.log_local / "rounds" / str(round_num) / FILE_RESULTS
         results_file.write_text(json.dumps(stats.to_dict(), indent=2))
 
+        self._save()
+
     def run_edit_phase(self, round_num: int) -> None:
         """Execute a single training round."""
         # Copy log to agent environments
@@ -100,6 +102,7 @@ def run_edit_phase(self, round_num: int) -> None:
                     self.logger.critical(f"Agent execution failed: {e}", exc_info=True)
                     raise
 
+        self._save()
         self.logger.info("Round completed.")
 
     def run_agent(self, agent: Player, round_num: int) -> None:
@@ -108,7 +111,10 @@ def run_agent(self, agent: Player, round_num: int) -> None:
         agent.run()
         agent.post_run_hook(round=round_num)
 
+    def _save(self) -> None:
+        (self.local_output_dir / "metadata.json").write_text(json.dumps(self.get_metadata(), indent=2))
+
     def end(self) -> None:
         """Save output files, clean up game resources and push agents if requested."""
-        (self.local_output_dir / "metadata.json").write_text(json.dumps(self.get_metadata(), indent=2))
+        self._save()
         self.game.end(self.cleanup_on_end)

codeclash/tournaments/single_player.py

@@ -102,6 +102,8 @@ def run_training_round(self, round_num: int) -> None:
         mirror_agent_state = round_num - 1 if round_num > 1 else 0
         self.set_mirror_state_to_round(mirror_agent_state)
 
+        self._save()
+
         self.logger.info("Round completed.")
 
     def run_main_agent(self, round_num: int):
@@ -116,10 +118,12 @@ def set_mirror_state_to_round(self, round_num: int):
         full_diff = filter_git_diff(full_diff)
         self.mirror_agent.reset_and_apply_patch(full_diff)
 
+    def _save(self) -> None:
+        (self.local_output_dir / "metadata.json").write_text(json.dumps(self.get_metadata(), indent=2))
+
     def end(self):
         """Clean up game resources."""
-        with open(self.local_output_dir / "metadata.json", "w") as f:
-            json.dump(self.get_metadata(), fp=f, indent=2)
+        self._save()
         self.game.end(self.cleanup_on_end)
 
     def evaluate(self, n_repetitions: int = 3) -> None: