From 844f0cdbc9272f8d465708820c8814076025781a Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Wed, 2 Oct 2019 22:49:00 -0400
Subject: [PATCH] fixed bug in tree observation.

---
 flatland/envs/observations.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 1edf943b..b258519b 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -280,7 +280,7 @@ class TreeObsForRailEnv(ObservationBuilder):
                         min_fractional_speed = current_fractional_speed
 
                     # Other direction agents
-                    # TODO: This does not work as expected yet
+                    # TODO: Test that this behavior is as expected
                     other_agent_opposite_direction += self.location_has_agent[position] - \
                                                       self.location_has_agent_direction[
                                                           (position, direction)]
-- 
GitLab