From 043e832db5bd2c5fabc33e6c977a62ba310eda68 Mon Sep 17 00:00:00 2001
From: Adrian Egli <adrian.egli@sbb.ch>
Date: Sat, 7 Nov 2020 00:01:15 +0100
Subject: [PATCH] .

---
 run.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/run.py b/run.py
index aaa4bab..d610fab 100644
--- a/run.py
+++ b/run.py
@@ -137,9 +137,9 @@ while True:
                             action = agent_last_action[agent]
                             nb_hit += 1
                         else:
-                            action = policy.act(observation[agent], eps=0.0)
-                            #if observation[agent][26] == 1:
-                            #    action = RailEnvActions.STOP_MOVING
+                            action = policy.act(observation[agent], eps=0.01)
+                            if observation[agent][26] == 1:
+                                action = RailEnvActions.STOP_MOVING
 
                         action_dict[agent] = action
 
-- 
GitLab