From dc8ab3bcdd12279d47796c7085977acfb7ecb7b3 Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Thu, 11 Jul 2019 15:40:13 -0400
Subject: [PATCH] refactored how we import envs by moving it into a generator
---
flatland/envs/generators.py | 2 +-
tests/test_file_load.py | 4 ++--
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py
index ff21046..907b4a2 100644
--- a/flatland/envs/generators.py
+++ b/flatland/envs/generators.py
@@ -197,7 +197,7 @@ def rail_from_manual_specifications_generator(rail_spec):
return generator
-def rail_from_data(filename):
+def rail_from_file(filename):
"""
Utility to load pickle file
diff --git a/tests/test_file_load.py b/tests/test_file_load.py
index 2b929b1..57fa45c 100644
--- a/tests/test_file_load.py
+++ b/tests/test_file_load.py
@@ -3,7 +3,7 @@
import numpy as np
-from flatland.envs.generators import rail_from_GridTransitionMap_generator, rail_from_data
+from flatland.envs.generators import rail_from_GridTransitionMap_generator, rail_from_file
from flatland.envs.observations import TreeObsForRailEnv
from flatland.envs.predictions import ShortestPathPredictorForRailEnv
from flatland.envs.rail_env import RailEnv
@@ -26,7 +26,7 @@ def test_load_pkl():
env = RailEnv(width=1,
height=1,
- rail_generator=rail_from_data(file_name),
+ rail_generator=rail_from_file(file_name),
number_of_agents=1,
obs_builder_object=TreeObsForRailEnv(max_depth=2, predictor=ShortestPathPredictorForRailEnv()),
)
--
GitLab