From ceaf338e412f7a8133d41b926a829fdf1efce4a3 Mon Sep 17 00:00:00 2001
From: Mattias Ljungstrom <mattias.ljungstrom@gmail.com>
Date: Sun, 28 Apr 2019 18:09:39 +0200
Subject: [PATCH] level gen: tweaks
---
examples/play_model.py | 4 ++--
flatland/envs/rail_env.py | 1 +
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/examples/play_model.py b/examples/play_model.py
index c51c8972..1f192fba 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 8132d933..0f15b3ab 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:
--
GitLab