From d62b21e28336eea76228f5d856e1b8e0d86a163d Mon Sep 17 00:00:00 2001 From: Matteo Bettini Date: Fri, 4 Oct 2024 10:23:16 +0100 Subject: [PATCH] disable single agent logging --- benchmarl/experiment/logger.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/benchmarl/experiment/logger.py b/benchmarl/experiment/logger.py index f1b15717..774cfaba 100644 --- a/benchmarl/experiment/logger.py +++ b/benchmarl/experiment/logger.py @@ -104,7 +104,12 @@ def log_collection( ) for group in self.group_map.keys(): group_episode_rewards = self._log_individual_and_group_rewards( - group, batch, gobal_done, any_episode_ended, to_log + group, + batch, + gobal_done, + any_episode_ended, + to_log, + log_individual_agents=False, # Turn on if you want single agent granularity ) # group_episode_rewards has shape (n_episodes) as we took the mean over agents in the group groups_episode_rewards.append(group_episode_rewards)