From aeada43445390483221f300b7d14fc764807cacb Mon Sep 17 00:00:00 2001
From: u214892 <u214892@sbb.ch>
Date: Fri, 5 Jul 2019 20:10:37 +0200
Subject: [PATCH] #62 increase coverage #83 cleanup
---
flatland/envs/rail_env.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index b31be94..d2b4791 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -202,7 +202,7 @@ class RailEnv(Environment):
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_agent, agent in enumerate(self.get_num_agents()):
+ for i_agent, agent in enumerate(self.agents):
agent.old_direction = agent.direction
agent.old_position = agent.position
if self.dones[i_agent]: # this agent has already completed...
--
GitLab