From ea0d33a2f0b110ccdc03d78a62e854062d72e4f3 Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Fri, 7 Jun 2019 15:12:28 +0200
Subject: [PATCH] Changed behavior of environment to change action_dict of
agents after their input. --> Giacomo please check that this is ok with your
implementation
---
flatland/envs/rail_env.py | 11 +++++++----
1 file changed, 7 insertions(+), 4 deletions(-)
diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index 410ee397..1b57ce59 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -199,7 +199,8 @@ class RailEnv(Environment):
if iAgent not in action_dict: # no action has been supplied for this agent
if agent.moving:
# Keep moving
- action_dict[iAgent] = RailEnvActions.MOVE_FORWARD
+ # CHange MOVE_FORWARD to DO_NOTHING
+ action_dict[iAgent] = RailEnvActions.DO_NOTHING
else:
action_dict[iAgent] = RailEnvActions.DO_NOTHING
@@ -214,12 +215,14 @@ class RailEnv(Environment):
if action == RailEnvActions.DO_NOTHING and agent.moving:
# Keep moving
- action_dict[iAgent] = RailEnvActions.MOVE_FORWARD
+ # Changed MOVE_FORWARD to DO_NOTHING
+ action_dict[iAgent] = RailEnvActions.DO_NOTHING
action = RailEnvActions.MOVE_FORWARD
if action == RailEnvActions.STOP_MOVING and agent.moving:
- action_dict[iAgent] = RailEnvActions.DO_NOTHING
- action = RailEnvActions.DO_NOTHING
+ #action_dict[iAgent] = RailEnvActions.DO_NOTHING
+ # CHanged DO_NOTHING to STOP_MOVING
+ action = RailEnvActions.STOP_MOVING
agent.moving = False
self.rewards_dict[iAgent] += stop_penalty
--
GitLab