diff --git a/examples/play_model.py b/examples/play_model.py index c51c8972988b9bd76149b786183d05b023e834ef..1f192fba854d267e05a5e0e77a2cb05a3e717730 100644 --- a/examples/play_model.py +++ b/examples/play_model.py @@ -20,7 +20,7 @@ def main(render=True, delay=0.0): #transition_probability = [0.5, # empty cell - Case 0 # 1.0, # Case 1 - straight # 1.0, # Case 2 - simple switch - # 0.3, # Case 3 - diamond drossing + # 0.3, # Case 3 - diamond crossing # 0.5, # Case 4 - single slip # 0.5, # Case 5 - double slip # 0.2, # Case 6 - symmetrical @@ -28,7 +28,7 @@ def main(render=True, delay=0.0): # Example generate a random rail env = RailEnv(width=15, height=15, - rail_generator=complex_rail_generator(nr_start_goal=20), + rail_generator=complex_rail_generator(nr_start_goal=20, min_dist=5), number_of_agents=1) if render: diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py index 8132d93329cc46837f9e5b1fad86562284e20cc3..0f15b3ab2fd142327605d71047cf91a0bf872eb3 100644 --- a/flatland/envs/rail_env.py +++ b/flatland/envs/rail_env.py @@ -326,6 +326,7 @@ def complex_rail_generator(nr_start_goal=1, min_dist=2, max_dist=99999, seed=0): if check_all_dist(sg_new): break start_goal.append([start, goal]) + print("Created #", len(start_goal), "pairs") # step 3: for sg in start_goal: