diff --git a/tests/test_flatland_env_sparse_rail_generator.py b/tests/test_flatland_env_sparse_rail_generator.py index 7079a4f9e3b176aaad5398d128eea63c7f8adccd..49739297513496d1019048493f29c172a2faaac8 100644 --- a/tests/test_flatland_env_sparse_rail_generator.py +++ b/tests/test_flatland_env_sparse_rail_generator.py @@ -8,7 +8,7 @@ from flatland.utils.rendertools import RenderTool import numpy as np def test_realistic_rail_generator(): - for test_loop in range(20): + for test_loop in range(5): num_agents = np.random.randint(10,30) env = RailEnv(width=np.random.randint(40,80), height=np.random.randint(10,20), @@ -18,7 +18,7 @@ def test_realistic_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) - time.sleep(1) + time.sleep(2) env_renderer.close_window() def test_sparse_rail_generator():