From 3832c57e0203f932e083cc7543103ccd6d460fc5 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Thu, 12 Sep 2019 08:04:33 +0200
Subject: [PATCH] bug fixed
---
examples/Simple_Realistic_Railway_Generator.py | 16 ++++++++--------
1 file changed, 8 insertions(+), 8 deletions(-)
diff --git a/examples/Simple_Realistic_Railway_Generator.py b/examples/Simple_Realistic_Railway_Generator.py
index c231971b..2057cbc1 100644
--- a/examples/Simple_Realistic_Railway_Generator.py
+++ b/examples/Simple_Realistic_Railway_Generator.py
@@ -262,7 +262,7 @@ def realistic_rail_generator(num_cities=5,
station_slots[city_loop].append(connection[int(np.floor(len(connection) / 2))])
nodes_added.append(start_node)
nodes_added.append(end_node)
-
+
if print_out_info:
print("max nbr of station slots with given configuration is:", station_slots_cnt)
@@ -441,15 +441,15 @@ def realistic_rail_generator(num_cities=5,
for itrials in range(100):
print(itrials, "generate new city")
np.random.seed(int(time.time()))
- env = RailEnv(width=120,
- height=120,
- rail_generator=realistic_rail_generator(num_cities=10,
- city_size=20,
- allowed_rotation_angles=[-90,0,90],
+ env = RailEnv(width=20+np.random.choice(100),
+ height=20+np.random.choice(100),
+ rail_generator=realistic_rail_generator(num_cities=2+np.random.choice(10),
+ city_size=4+np.random.choice(20),
+ allowed_rotation_angles=[-90,-30,0,30,90],
max_number_of_station_tracks=4,
nbr_of_switches_per_station_track=2,
- max_number_of_connecting_tracks=1,
- do_random_connect_stations=True,
+ max_number_of_connecting_tracks=3,
+ do_random_connect_stations=False,
# Number of cities in map
seed=int(time.time()) # Random seed
),
--
GitLab