Skip to content

Commit

Permalink
Umshini: fix minor bug with printing and render_mode (#119)
Browse files Browse the repository at this point in the history
  • Loading branch information
elliottower authored Dec 21, 2023
1 parent 2f4f5b5 commit 7367b05
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion chatarena/environments/umshini/pettingzoo_wrapper.py
Original file line number Diff line number Diff line change
Expand Up @@ -590,7 +590,7 @@ def step(self, action: str):
self.render()

# Print final scores if the env has just terminated (debate moderator final message already shows scores)
if termination and self.env_name != "debate":
if termination and self.env_name != "debate" and self.render_mode is not None:
print(Fore.BLACK + f"TOTAL SCORES: {self.total_rewards}")

# Get the next agent in PettingZoo, and iterate the underlying environment (used for reward calculations)
Expand Down

0 comments on commit 7367b05

Please sign in to comment.