diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py index 34e9184592ffa0bd768f8477875dbbb662e4fc07..4ada546585633bb82f7d0b7459e6909158f96131 100644 --- a/flatland/envs/generators.py +++ b/flatland/envs/generators.py @@ -687,13 +687,6 @@ def realistic_rail_generator(nr_start_goal=1, seed=0): goal_track = (off_set, width - 1) new_path = connect_rail(rail_trans, rail_array, start_track, goal_track) - - add_pos = (int((start_track[0] + goal_track[0]) / 2), int((start_track[1] + goal_track[1]) / 2)) - agents_positions.append(add_pos) - agents_directions.append(np.random.choice([3, 1])) - add_pos = (int((start_track[0] + goal_track[0]) / 2), int((2 * start_track[1] + goal_track[1]) / 3)) - agents_targets.append(add_pos) - # track one (full track : left right) if off_set_loop > 0: if off_set_loop % 2 == 1: @@ -758,12 +751,11 @@ def realistic_rail_generator(nr_start_goal=1, seed=0): c = (off_set + nbr_track_loop, x[2 * x_loop + 1] + 1) make_switch_w_e(width, height, grid_map, c) - if nbr_track_loop > 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])) - add_pos = (int((start[0] + goal[0]) / 2), int((2*start[1] + goal[1]) / 3)) - agents_targets.append(add_pos) + 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])) + add_pos = (int((start[0] + goal[0]) / 2), int((2*start[1] + goal[1]) / 3)) + agents_targets.append(add_pos) agents_position = [] agents_target = []