Merge pull request #75 from elliottower/umshini-minor-bugfix-rendering

Umshini: minor rendering bugfix for local environments
Farama-Foundation · Nov 8, 2023 · ab305f6 · ab305f6
2 parents 7130aa7 + d88ad8d
commit ab305f6
Showing 1 changed file with 0 additions and 5 deletions.
diff --git 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