From da6d4606e51d043934eb6e2ff12fc441d3653ba3 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Tue, 20 Aug 2019 14:59:42 +0200
Subject: [PATCH] realistic generator
---
flatland/envs/generators.py | 8 ++++----
tests/test_flatland_env_sparse_rail_generator.py | 2 +-
2 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py
index d21c6eb1..aba43037 100644
--- a/flatland/envs/generators.py
+++ b/flatland/envs/generators.py
@@ -847,16 +847,16 @@ def realistic_rail_generator(nr_start_goal=1, seed=0, add_max_dead_end=4, two_tr
for off_set_loop in range(len(x_offsets)):
off_set = x_offsets[off_set_loop]
- pos_ys = np.random.choice(np.arange(width - 7) + 3, min(width - 7, add_max_dead_end), False)
+ pos_ys = np.random.choice(np.arange(width - 7) + 4, min(width - 7, add_max_dead_end), False)
for pos_y in pos_ys:
pos_x = off_set + 1 + int(two_track_back_bone)
if pos_x < height - 1:
ok = True
- for k in range(4):
+ for k in range(5):
if two_track_back_bone:
- c = (pos_x - 1, pos_y - k + 1)
+ c = (pos_x - 1, pos_y - k + 2)
ok &= grid_map.grid[c[0]][c[1]] == 1025
- c = (pos_x, pos_y - k + 1)
+ c = (pos_x, pos_y - k + 2)
ok &= grid_map.grid[c[0]][c[1]] == 0
if ok:
if np.random.random() < 0.5:
diff --git a/tests/test_flatland_env_sparse_rail_generator.py b/tests/test_flatland_env_sparse_rail_generator.py
index 710ae6d5..94b42149 100644
--- a/tests/test_flatland_env_sparse_rail_generator.py
+++ b/tests/test_flatland_env_sparse_rail_generator.py
@@ -53,4 +53,4 @@ def test_sparse_rail_generator():
env_renderer.render_env(show=True, show_observations=True, show_predictions=False)
env_renderer.close_window()
-test_realistic_rail_generator("./../rendering/")
+# test_realistic_rail_generator("./../rendering/")
--
GitLab