From 357be98eed388151b48ba9dbb8a2d456d333c033 Mon Sep 17 00:00:00 2001 From: MLErik <baerenjesus@gmail.com> Date: Mon, 19 Aug 2019 11:32:13 -0400 Subject: [PATCH] fixed formatting --- tests/test_flatland_env_sparse_rail_generator.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/test_flatland_env_sparse_rail_generator.py b/tests/test_flatland_env_sparse_rail_generator.py index 86e02c75..62937213 100644 --- a/tests/test_flatland_env_sparse_rail_generator.py +++ b/tests/test_flatland_env_sparse_rail_generator.py @@ -15,10 +15,12 @@ def test_realistic_rail_generator(vizualization_folder_name=None): number_of_agents=num_agents, obs_builder_object=GlobalObsForRailEnv()) # 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) env_renderer.render_env(show=True, show_observations=True, show_predictions=False) + def test_sparse_rail_generator(): env = RailEnv(width=50, height=50, @@ -36,4 +38,3 @@ def test_sparse_rail_generator(): # reset to initialize agents_static env_renderer = RenderTool(env, gl="PILSVG", ) env_renderer.render_env(show=True, show_observations=True, show_predictions=False) - -- GitLab