diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py index 3135e9e4979f6e170654bec615d2fe5475ab0bf4..40e4ab7e16217a90ba1101b52a1184aa26a53b31 100644 --- a/flatland/envs/rail_env.py +++ b/flatland/envs/rail_env.py @@ -180,6 +180,7 @@ class RailEnv(Environment): self.rewards_dict[iAgent] = 0 if self.dones["__all__"]: + self.rewards_dict = [r + global_reward for r in self.rewards_dict] return self._get_observations(), self.rewards_dict, self.dones, {} # for i in range(len(self.agents_handles)):