diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py index d21c6eb1d01e3cb0e9296e6e097d0d944592d2e4..aba4303739494a6fef5305158fde60c7f972fb60 100644 --- a/flatland/envs/generators.py +++ b/flatland/envs/generators.py @@ -847,16 +847,16 @@ def realistic_rail_generator(nr_start_goal=1, seed=0, add_max_dead_end=4, two_tr for off_set_loop in range(len(x_offsets)): off_set = x_offsets[off_set_loop] - pos_ys = np.random.choice(np.arange(width - 7) + 3, min(width - 7, add_max_dead_end), False) + pos_ys = np.random.choice(np.arange(width - 7) + 4, min(width - 7, add_max_dead_end), False) for pos_y in pos_ys: pos_x = off_set + 1 + int(two_track_back_bone) if pos_x < height - 1: ok = True - for k in range(4): + for k in range(5): if two_track_back_bone: - c = (pos_x - 1, pos_y - k + 1) + c = (pos_x - 1, pos_y - k + 2) ok &= grid_map.grid[c[0]][c[1]] == 1025 - c = (pos_x, pos_y - k + 1) + c = (pos_x, pos_y - k + 2) ok &= grid_map.grid[c[0]][c[1]] == 0 if ok: if np.random.random() < 0.5: diff --git a/tests/test_flatland_env_sparse_rail_generator.py b/tests/test_flatland_env_sparse_rail_generator.py index 710ae6d54ad89cc26a01bcaf59557e198196f5c1..94b42149377f2efeffb891deb49d4c8b0df4efd5 100644 --- a/tests/test_flatland_env_sparse_rail_generator.py +++ b/tests/test_flatland_env_sparse_rail_generator.py @@ -53,4 +53,4 @@ def test_sparse_rail_generator(): env_renderer.render_env(show=True, show_observations=True, show_predictions=False) env_renderer.close_window() -test_realistic_rail_generator("./../rendering/") +# test_realistic_rail_generator("./../rendering/")