From 6db70ca5b53bda2a65416391823fb2f419d18e8b Mon Sep 17 00:00:00 2001
From: u214892 <u214892@sbb.ch>
Date: Fri, 5 Jul 2019 16:35:45 +0200
Subject: [PATCH] #67 multiple agents

---
 tests/test_flatland_envs_observations.py | 2 +-
 tests/test_flatland_envs_predictions.py  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_flatland_envs_observations.py b/tests/test_flatland_envs_observations.py
index 5ee5b4a..fb45e5f 100644
--- a/tests/test_flatland_envs_observations.py
+++ b/tests/test_flatland_envs_observations.py
@@ -6,7 +6,7 @@ import numpy as np
 from flatland.envs.generators import rail_from_GridTransitionMap_generator
 from flatland.envs.observations import GlobalObsForRailEnv
 from flatland.envs.rail_env import RailEnv
-from simple_rail import make_simple_rail
+from tests.simple_rail import make_simple_rail
 
 """Tests for `flatland` package."""
 
diff --git a/tests/test_flatland_envs_predictions.py b/tests/test_flatland_envs_predictions.py
index 2926d21..96a4b95 100644
--- a/tests/test_flatland_envs_predictions.py
+++ b/tests/test_flatland_envs_predictions.py
@@ -10,7 +10,7 @@ from flatland.envs.observations import TreeObsForRailEnv
 from flatland.envs.predictions import DummyPredictorForRailEnv, ShortestPathPredictorForRailEnv
 from flatland.envs.rail_env import RailEnv
 from flatland.utils.rendertools import RenderTool
-from simple_rail import make_simple_rail
+from tests.simple_rail import make_simple_rail
 
 """Test predictions for `flatland` package."""
 
-- 
GitLab