Skip to content
Snippets Groups Projects
Commit 21da2117 authored by Erik Nygren's avatar Erik Nygren
Browse files

added jeremys level generator to training file for fun ;)

parent c69ceb5e
No related branches found
No related tags found
No related merge requests found
...@@ -23,7 +23,14 @@ transition_probability = [5, # empty cell - Case 0 ...@@ -23,7 +23,14 @@ transition_probability = [5, # empty cell - Case 0
env = RailEnv(width=10, env = RailEnv(width=10,
height=10, height=10,
rail_generator=random_rail_generator(cell_type_relative_proportion=transition_probability), 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") env_renderer = RenderTool(env, gl="QT")
handle = env.get_agent_handles() handle = env.get_agent_handles()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment