From 12b18d048c73f66825e307bbf57b78bb29d00665 Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Thu, 16 May 2019 17:54:03 +0200
Subject: [PATCH] updated test_environments.py to import correct library
---
flatland/envs/observations.py | 3 +++
tests/test_environments.py | 2 +-
2 files changed, 4 insertions(+), 1 deletion(-)
diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 4fad2c0..bde43bf 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -396,6 +396,9 @@ class TreeObsForRailEnv(ObservationBuilder):
observation = observation + branch_observation
if len(branch_visited) != 0:
visited.union(branch_visited)
+ print(visited)
+ print(branch_visited)
+ print("-------")
else:
num_cells_to_fill_in = 0
pow4 = 1
diff --git a/tests/test_environments.py b/tests/test_environments.py
index 4c55eac..9c7b53b 100644
--- a/tests/test_environments.py
+++ b/tests/test_environments.py
@@ -7,7 +7,7 @@ from flatland.envs.generators import rail_from_GridTransitionMap_generator
from flatland.envs.generators import complex_rail_generator
from flatland.core.transitions import Grid4Transitions
from flatland.core.transition_map import GridTransitionMap
-from flatland.core.env_observation_builder import GlobalObsForRailEnv
+from flatland.envs.observations import GlobalObsForRailEnv
from flatland.envs.agent_utils import EnvAgent
"""Tests for `flatland` package."""
--
GitLab