From 379b7e48577b3cc6dc6a411410fcaa1a6707c25f Mon Sep 17 00:00:00 2001
From: hagrid67 <jdhwatson@gmail.com>
Date: Tue, 21 May 2019 21:22:10 +0100
Subject: [PATCH] Added debug / error line in observations.py for editor
---
flatland/envs/observations.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index fa77b58a..ba159cb7 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -195,7 +195,8 @@ class TreeObsForRailEnv(ObservationBuilder):
# for loc in self.env.agents_position:
# self.location_has_agent[(loc[0], loc[1])] = 1
self.location_has_agent = {tuple(agent.position): 1 for agent in self.env.agents}
-
+ if handle > len(self.env.agents):
+ print("ERROR: obs _get - handle ", handle, " len(agents)", len(self.env.agents))
agent = self.env.agents[handle] # TODO: handle being treated as index
# position = self.env.agents_position[handle]
# orientation = self.env.agents_direction[handle]
--
GitLab