From d925dfb7083060e2e4628676cc6a515b2d0a5bca Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Wed, 2 Oct 2019 17:55:29 -0400
Subject: [PATCH] need to update tree observation as well
---
flatland/envs/observations.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 548c3e29..65708e5f 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -161,6 +161,7 @@ class TreeObsForRailEnv(ObservationBuilder):
# Update local lookup table for all agents' positions
self.location_has_agent = {tuple(agent.position): 1 for agent in self.env.agents}
+ # TODO: Update this to handle number of agents at same location
self.location_has_agent_direction = {tuple(agent.position): agent.direction for agent in self.env.agents}
self.location_has_agent_speed = {tuple(agent.position): agent.speed_data['speed'] for agent in self.env.agents}
self.location_has_agent_malfunction = {tuple(agent.position): agent.malfunction_data['malfunction'] for agent in
--
GitLab