#67 multiple agents

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."""
 

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."""