From d6bd0d9fdf70a82fd2ed69024ce2bbfecb5b5b1b Mon Sep 17 00:00:00 2001 From: elliottower Date: Thu, 21 Dec 2023 14:15:29 -0500 Subject: [PATCH] Fix minor bug in SCORES being printed even with render mode None (umshini client) --- chatarena/environments/umshini/pettingzoo_wrapper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/chatarena/environments/umshini/pettingzoo_wrapper.py b/chatarena/environments/umshini/pettingzoo_wrapper.py index deb7cc6f..45c52350 100644 --- a/chatarena/environments/umshini/pettingzoo_wrapper.py +++ b/chatarena/environments/umshini/pettingzoo_wrapper.py @@ -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)