From 01ced7031580b0cbbfb4bbf8f4b7c3f06fcced97 Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Thu, 13 Jun 2019 13:40:03 +0200
Subject: [PATCH] bug fix

---
 flatland/envs/observations.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 23a95ad..7bc2f99 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -328,8 +328,6 @@ class TreeObsForRailEnv(ObservationBuilder):
                     # Cummulate the number of agents on branch with other direction
                     other_agent_opposite_direction += 1
 
-            if self.predictor:
-                continue
             # Register possible conflict
 
             if position in self.location_has_target:
-- 
GitLab