diff --git a/tests/test_flatland_envs_rail_env.py b/tests/test_flatland_envs_rail_env.py index 5c518a27160f28a8a3518a4fe013f97324e4d41e..351a326a19c078e7f525c4b96ad8662777f7dd2d 100644 --- a/tests/test_flatland_envs_rail_env.py +++ b/tests/test_flatland_envs_rail_env.py @@ -80,7 +80,7 @@ def test_rail_environment_single_agent(): schedule_generator=random_schedule_generator(), number_of_agents=1, obs_builder_object=GlobalObsForRailEnv()) - for _ in range(200): + for _ in range(50): _ = rail_env.reset(False, False, True) # We do not care about target for the moment