diff --git a/examples/flatland_2_0_example.py b/examples/flatland_2_0_example.py index 57e99ffb89e9f6584fff9707de921c5fda101585..6396f4123adf895c381c2a21f6d8dc6e4e823b92 100644 --- a/examples/flatland_2_0_example.py +++ b/examples/flatland_2_0_example.py @@ -32,8 +32,8 @@ speed_ration_map = {1.: 0.25, # Fast passenger train env = RailEnv(width=50, height=50, - rail_generator=sparse_rail_generator(num_cities=50, # Number of cities in map (where train stations are) - seed=0, # Random seed + rail_generator=sparse_rail_generator(num_cities=10, # Number of cities in map (where train stations are) + seed=1, # Random seed grid_mode=False, max_inter_city_rails=2, max_tracks_in_city=4, diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py index b5bc44f2e698a4ffa23ca31d34ac14f613340d04..901f318c0add611c4d0ebb0846c7a4c0c30f39dd 100644 --- a/flatland/envs/rail_env.py +++ b/flatland/envs/rail_env.py @@ -248,6 +248,7 @@ class RailEnv(Environment): rc_pos = (r, c) check = self.rail.cell_neighbours_valid(rc_pos, True) if not check: + print(self.rail.grid[rc_pos]) warnings.warn("Invalid grid at {} -> {}".format(rc_pos, check)) # TODO https://gitlab.aicrowd.com/flatland/flatland/issues/172 # hacky: we must re-compute the distance map and not use the initial distance_map loaded from file by