disable single agent logging

facebookresearch · Oct 4, 2024 · d62b21e · d62b21e
1 parent ab30f13
commit d62b21e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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)