From 21da2117af441965fe949f1a58a24fbd24fe29cd Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Fri, 26 Apr 2019 11:37:01 +0200
Subject: [PATCH] added jeremys level generator to training file for fun ;)
---
examples/training_navigation.py | 9 ++++++++-
1 file changed, 8 insertions(+), 1 deletion(-)
diff --git a/examples/training_navigation.py b/examples/training_navigation.py
index a7b2fbd..4cd62ee 100644
--- a/examples/training_navigation.py
+++ b/examples/training_navigation.py
@@ -23,7 +23,14 @@ transition_probability = [5, # empty cell - Case 0
env = RailEnv(width=10,
height=10,
rail_generator=random_rail_generator(cell_type_relative_proportion=transition_probability),
- number_of_agents=5)
+ number_of_agents=2)
+
+env = RailEnv(width=20,
+ height=20,
+ rail_generator=rail_from_list_of_saved_GridTransitionMap_generator(
+ ['../flatland/baselines/test-editor.npy']),
+ number_of_agents=1)
+
env_renderer = RenderTool(env, gl="QT")
handle = env.get_agent_handles()
--
GitLab