From bdfd4f09b0078a8c5d601f7cc75e375c9ca2c283 Mon Sep 17 00:00:00 2001
From: Michel Marti <michel.marti@t-systems.com>
Date: Tue, 12 Oct 2021 16:46:01 +0200
Subject: [PATCH] changed 'status' to 'state'

---
 flatland/contrib/wrappers/flatland_wrappers.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/flatland/contrib/wrappers/flatland_wrappers.py b/flatland/contrib/wrappers/flatland_wrappers.py
index 6e4772b9..d07fd06a 100644
--- a/flatland/contrib/wrappers/flatland_wrappers.py
+++ b/flatland/contrib/wrappers/flatland_wrappers.py
@@ -25,7 +25,7 @@ def possible_actions_sorted_by_distance(env: RailEnv, handle: int):
         agent_virtual_position = agent.position
     else:
         print("no action possible!")
-        print("agent status: ", agent.state)
+        print("agent state: ", agent.state)
         # NEW: if agent is at target, DO_NOTHING, and distance is zero.
         # NEW: (needs to be tested...)
         return [(RailEnvActions.DO_NOTHING, 0)] * 2
@@ -215,7 +215,7 @@ class NoChoiceCellsSkipper:
         i["action_required"] = dict()
         i["malfunction"] = dict()
         i["speed"] = dict()
-        i["status"] = dict() # TODO: change to "state"
+        i["state"] = dict()
 
         while len(o) == 0:
             obs, reward, done, info = self.env.step(action_dict)
@@ -231,7 +231,7 @@ class NoChoiceCellsSkipper:
                     i["action_required"][agent_id] = info["action_required"][agent_id] 
                     i["malfunction"][agent_id] = info["malfunction"][agent_id]
                     i["speed"][agent_id] = info["speed"][agent_id]
-                    i["status"][agent_id] = info["status"][agent_id] # TODO: change to "state"
+                    i["state"][agent_id] = info["state"][agent_id]
                                                                   
                     if self.accumulate_skipped_rewards:
                         discounted_skipped_reward = r[agent_id]
-- 
GitLab