diff --git a/chatarena/environments/umshini/pettingzoo_wrapper.py b/chatarena/environments/umshini/pettingzoo_wrapper.py index a5ac2d85..a6600b5d 100644 --- a/chatarena/environments/umshini/pettingzoo_wrapper.py +++ b/chatarena/environments/umshini/pettingzoo_wrapper.py @@ -470,11 +470,6 @@ def step(self, action: str): observation, reward, termination, truncation, info = self._unravel_timestep( timestep ) - # add moderator messages to info so they are rendered - # some environments (e.g., debate) have the moderator announce the winner as the last message - if termination or truncation: - if info["all_messages"][-1].agent_name == "Moderator": - info["new_messages"].append(info["all_messages"][-2]) self.observations[agent] = observation self.rewards = reward