Skip to content
Snippets Groups Projects
Commit 357be98e authored by Erik Nygren's avatar Erik Nygren :bullettrain_front:
Browse files

fixed formatting

parent a9716d8a
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
tests/test_flatland_env_sparse_rail_generator.py
+ 3
− 2
View file @ 357be98e
...@@ -15,10 +15,12 @@ def test_realistic_rail_generator(vizualization_folder_name=None): ...@@ -15,10 +15,12 @@ def test_realistic_rail_generator(vizualization_folder_name=None):
number_of_agents=num_agents, number_of_agents=num_agents,
obs_builder_object=GlobalObsForRailEnv()) obs_builder_object=GlobalObsForRailEnv())
# reset to initialize agents_static # reset to initialize agents_static
env_renderer = RenderTool(env, gl="PILSVG", agent_render_variant=AgentRenderVariant.ONE_STEP_BEHIND, screen_height=1200, env_renderer = RenderTool(env, gl="PILSVG", agent_render_variant=AgentRenderVariant.ONE_STEP_BEHIND,
screen_height=1200,
screen_width=1600) screen_width=1600)
env_renderer.render_env(show=True, show_observations=True, show_predictions=False) env_renderer.render_env(show=True, show_observations=True, show_predictions=False)
def test_sparse_rail_generator(): def test_sparse_rail_generator():
env = RailEnv(width=50, env = RailEnv(width=50,
height=50, height=50,
...@@ -36,4 +38,3 @@ def test_sparse_rail_generator(): ...@@ -36,4 +38,3 @@ def test_sparse_rail_generator():
# reset to initialize agents_static # reset to initialize agents_static
env_renderer = RenderTool(env, gl="PILSVG", ) env_renderer = RenderTool(env, gl="PILSVG", )
env_renderer.render_env(show=True, show_observations=True, show_predictions=False) env_renderer.render_env(show=True, show_observations=True, show_predictions=False)
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment