From afb0555b9a5f982fce1c705b46f6803dd2cf749c Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Mon, 30 Sep 2019 16:36:14 -0400
Subject: [PATCH] greater min distance between cities
---
examples/flatland_2_0_example.py | 4 ++--
flatland/envs/rail_generators.py | 2 +-
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/examples/flatland_2_0_example.py b/examples/flatland_2_0_example.py
index 6396f412..57e99ffb 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=10, # Number of cities in map (where train stations are)
- seed=1, # Random seed
+ rail_generator=sparse_rail_generator(num_cities=50, # Number of cities in map (where train stations are)
+ seed=0, # Random seed
grid_mode=False,
max_inter_city_rails=2,
max_tracks_in_city=4,
diff --git a/flatland/envs/rail_generators.py b/flatland/envs/rail_generators.py
index 6cfbece9..fcd5e672 100644
--- a/flatland/envs/rail_generators.py
+++ b/flatland/envs/rail_generators.py
@@ -632,7 +632,7 @@ def sparse_rail_generator(num_cities=5, grid_mode=False, max_inter_city_rails=4,
to_close = False
# Check distance to nodes
for node_pos in node_positions:
- if _city_overlap((x_tmp, y_tmp), node_pos, 2 * node_radius + 2):
+ if _city_overlap((x_tmp, y_tmp), node_pos, 2 * (node_radius + 1) + 1):
to_close = True
if not to_close:
--
GitLab