diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py index 4ada546585633bb82f7d0b7459e6909158f96131..d63e161ae48712b17433610709a9c4ef54cb237d 100644 --- a/flatland/envs/generators.py +++ b/flatland/envs/generators.py @@ -753,7 +753,7 @@ def realistic_rail_generator(nr_start_goal=1, seed=0): add_pos = (int((start[0] + goal[0]) / 2), int((start[1] + goal[1]) / 2)) agents_positions.append(add_pos) - agents_directions.append(np.random.choice([3, 1])) + agents_directions.append(3) add_pos = (int((start[0] + goal[0]) / 2), int((2*start[1] + goal[1]) / 3)) agents_targets.append(add_pos)