diff --git a/core/population.py b/core/population.py index 2e4e00a..151a368 100644 --- a/core/population.py +++ b/core/population.py @@ -818,9 +818,9 @@ def compute_system_rewards(self) -> Dict[str, float]: agent_reward = agent.compute_morphology_statistics() for k, v in agent_reward.items(): reward[k] += v - # average - for k in agent_reward.keys(): - agent_reward[k] = agent_reward[k] / len(self.agents) + # average + for k in reward.keys(): + reward[k] = reward[k] / len(self.agents) # Pairwise disagreement across agents' forms disagr = 0