From 4ad3eae7d0e18a973f6f65564051657bf90caed3 Mon Sep 17 00:00:00 2001
From: u214892 <u214892@sbb.ch>
Date: Wed, 4 Sep 2019 16:32:07 +0200
Subject: [PATCH] #167 bugfix action_on_cellexit

---
 flatland/envs/rail_env.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index 566fa7d1..6f4b037e 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -342,6 +342,7 @@ class RailEnv(Environment):
                 if agent.malfunction_data['malfunction'] < 2:
                     agent.malfunction_data['malfunction'] -= 1
                     self.agents[i_agent].moving = True
+                    action_dict[i_agent] = RailEnvActions.DO_NOTHING
 
                 else:
                     agent.malfunction_data['malfunction'] -= 1
-- 
GitLab