From 11a275d1b780935980855fa430617d08b869d7f9 Mon Sep 17 00:00:00 2001
From: Giacomo Spigler <spiglerg@gmail.com>
Date: Tue, 2 Jul 2019 09:11:12 +0200
Subject: [PATCH] fixed issue #73
---
flatland/envs/rail_env.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index 3fe1edb..17e8827 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -189,7 +189,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]
+ self.rewards_dict = {i:r + global_reward for i,r in self.rewards_dict.items()}
return self._get_observations(), self.rewards_dict, self.dones, {}
# for i in range(len(self.agents_handles)):
@@ -297,7 +297,7 @@ class RailEnv(Environment):
# Check for end of episode + add global reward to all rewards!
if np.all([np.array_equal(agent2.position, agent2.target) for agent2 in self.agents]):
self.dones["__all__"] = True
- self.rewards_dict = [0 * r + global_reward for r in self.rewards_dict]
+ self.rewards_dict = {i:0 * r + global_reward for i,r in self.rewards_dict.items()}
return self._get_observations(), self.rewards_dict, self.dones, {}
--
GitLab