From 48f77501588a48ec8f113834ae26cfd6aadc4e58 Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Sat, 28 Sep 2019 16:06:26 -0400
Subject: [PATCH] issue with boarder resolved
---
examples/flatland_2_0_example.py | 2 +-
flatland/envs/rail_generators.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/examples/flatland_2_0_example.py b/examples/flatland_2_0_example.py
index feb37909..4a614b2f 100644
--- a/examples/flatland_2_0_example.py
+++ b/examples/flatland_2_0_example.py
@@ -30,7 +30,7 @@ speed_ration_map = {1.: 0.25, # Fast passenger train
env = RailEnv(width=50,
height=50,
- rail_generator=sparse_rail_generator(num_cities=9, # Number of cities in map (where train stations are)
+ rail_generator=sparse_rail_generator(num_cities=12, # Number of cities in map (where train stations are)
seed=0, # Random seed
grid_mode=False,
max_inter_city_rails=2,
diff --git a/flatland/envs/rail_generators.py b/flatland/envs/rail_generators.py
index a9e9796f..74d17b83 100644
--- a/flatland/envs/rail_generators.py
+++ b/flatland/envs/rail_generators.py
@@ -551,7 +551,7 @@ def sparse_rail_generator(num_cities=5, grid_mode=False, max_inter_city_rails=4,
rail_array = grid_map.grid
rail_array.fill(0)
np.random.seed(seed + num_resets)
- node_radius = int(max_tracks_in_city / 2) + 1
+ node_radius = int(np.ceil(max_tracks_in_city / 2))
max_inter_city_rails_allowed = max_inter_city_rails
if max_inter_city_rails_allowed > max_tracks_in_city:
max_inter_city_rails_allowed = max_tracks_in_city
--
GitLab