Commit aeada434 authored by u214892's avatar u214892
Browse files

#62 increase coverage #83 cleanup

parent 001b58c7
Pipeline #1352 failed with stage
in 10 minutes and 15 seconds
......@@ -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...
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment