diff --git a/tests/test_environments.py b/tests/test_environments.py index 435002b5af080293f4a739b12f6e5256c8780e78..d5e7dd9c9fc350f11482b684572858b34ac83829 100644 --- a/tests/test_environments.py +++ b/tests/test_environments.py @@ -75,7 +75,7 @@ def test_rail_environment_single_agent(): assert(initial_pos[0] == rail_env.agents_position[0][0]) # We check that the train always attains its target after some time - for _ in range(200): + for _ in range(10): _ = rail_env.reset() done = False