diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py index 1b57ce59e05b8d939323afaa8800cba8254d2201..8a2bcf49ff2301cb47ff74acada59ca5b8ccbef8 100644 --- a/flatland/envs/rail_env.py +++ b/flatland/envs/rail_env.py @@ -199,7 +199,7 @@ class RailEnv(Environment): if iAgent not in action_dict: # no action has been supplied for this agent if agent.moving: # Keep moving - # CHange MOVE_FORWARD to DO_NOTHING + # Change MOVE_FORWARD to DO_NOTHING action_dict[iAgent] = RailEnvActions.DO_NOTHING else: action_dict[iAgent] = RailEnvActions.DO_NOTHING @@ -216,13 +216,13 @@ class RailEnv(Environment): if action == RailEnvActions.DO_NOTHING and agent.moving: # Keep moving # Changed MOVE_FORWARD to DO_NOTHING - action_dict[iAgent] = RailEnvActions.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 # CHanged DO_NOTHING to STOP_MOVING - action = RailEnvActions.STOP_MOVING + # action = RailEnvActions.STOP_MOVING agent.moving = False self.rewards_dict[iAgent] += stop_penalty