Class RoundStats

codeclash/arenas/arena.py:40–69

        }


class RoundStats:
    def __init__(self, round_num: int, agents: list[Player]):
        self.winner = None
        self.round_num = round_num
        # Map of player to game metric (e.g. # of wins, assets accumulated)
        self.scores: dict[str, float] = {a.name: 0.0 for a in agents}
        self.player_stats: dict[str, PlayerStats] = {agent.name: PlayerStats(name=agent.name) for agent in agents}
        self.details: list[str] = []

    def __str__(self) -> str:
        rv = [f"In round {self.round_num}, the winner is {self.winner}.", "\nSummary of player performance:"]
        for player, stats in self.player_stats.items():
            if stats.valid_submit:
                rv.append(f"- {player}: submission compiled successfully, score={stats.score}")
            else:
                rv.append(f"- {player}: submission failed with error: {stats.invalid_reason}")
        if self.details:
            rv.extend(["Details:"] + [f"- {line}" for line in self.details])
        return "\n".join(rv)

    def to_dict(self) -> dict[str, Any]:
        # Going through some pain to ensure that the scores dict is always complete
        player_names = set(self.player_stats.keys()) | set(self.scores.keys())
        return {
            "round_num": self.round_num,
            "winner": self.winner,
            "details": self.details,
            "scores": {name: self.scores.get(name, 0.0) for name in player_names},
            "player_stats": {name: stats.to_dict() for name, stats in self.player_stats.items()},
        }