diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py index 2af0d8b962fc5b31e2e3c7607c06b8613c15a47a..abe623ae173a593e265cff7d4d88eb323e16b08e 100644 --- a/flatland/envs/rail_env.py +++ b/flatland/envs/rail_env.py @@ -334,6 +334,7 @@ class RailEnv(Environment): if np.equal(agent.position, agent.target).all(): self.dones[i_agent] = True + agent.moving = False else: self.rewards_dict[i_agent] += step_penalty * agent.speed_data['speed']