From 3528da79555b9dd2d5827f6e88412e63cea6fc25 Mon Sep 17 00:00:00 2001 From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch> Date: Fri, 16 Aug 2019 21:02:59 +0200 Subject: [PATCH] bug fix in generator --- flatland/envs/generators.py | 2 +- tests/test_flatland_env_sparse_rail_generator.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py index 272ae148..aef00cdd 100644 --- a/flatland/envs/generators.py +++ b/flatland/envs/generators.py @@ -733,7 +733,7 @@ def realistic_rail_generator(nr_start_goal=1, seed=0): (x_offsets[off_set_loop] - 1, 0), (x_offsets[off_set_loop] - 2, 0)) - for nbr_track_loop in range(max_n_track_seg-1): + for nbr_track_loop in range(max_n_track_seg-1): if len(data) < 2 * n_track_seg + 1: break x = np.sort(np.random.choice(data, 2 * n_track_seg, False)).astype(int) diff --git a/tests/test_flatland_env_sparse_rail_generator.py b/tests/test_flatland_env_sparse_rail_generator.py index 49739297..3c1bc18b 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(5): + for test_loop in range(20): num_agents = np.random.randint(10,30) env = RailEnv(width=np.random.randint(40,80), height=np.random.randint(10,20), -- GitLab