Update mettagrid_env.py

Metta-AI · Jan 23, 2025 · 07e8840 · 07e8840
1 parent 677fb75
commit 07e8840
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mettagrid/mettagrid_env.py b/mettagrid/mettagrid_env.py
@@ -79,7 +79,7 @@ def process_episode_stats(self, infos: Dict[str, Any]):
         })
         stats = self._c_env.get_episode_stats()
 
-        infos["episode_rewards"] = 1000 * episode_rewards / self._env_cfg.game.max_steps
+        infos["episode_rewards"] = episode_rewards
         infos["agent_raw"] = stats["agent"]
         infos["game"] = stats["game"]
         infos["agent"] = {}