From 5495385194afc503dd5c1301151dbf1528ecca6d Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Thu, 16 May 2019 17:37:49 +0200
Subject: [PATCH] updated test for observation to account for new structure in
 repo

---
 tests/test_env_observation_builder.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_env_observation_builder.py b/tests/test_env_observation_builder.py
index 1692a98..13f8d8f 100644
--- a/tests/test_env_observation_builder.py
+++ b/tests/test_env_observation_builder.py
@@ -3,7 +3,7 @@
 
 import numpy as np
 
-from flatland.core.env_observation_builder import GlobalObsForRailEnv
+from flatland.envs.observations import GlobalObsForRailEnv
 from flatland.core.transition_map import GridTransitionMap, Grid4Transitions
 from flatland.envs.rail_env import RailEnv
 from flatland.envs.generators import rail_from_GridTransitionMap_generator
-- 
GitLab