From e1168bb6b7e407290cfff38586302e3bd1c943a8 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Thu, 12 Sep 2019 19:12:34 +0200
Subject: [PATCH] refactored
---
examples/Simple_Realistic_Railway_Generator.py | 14 +++++++-------
1 file changed, 7 insertions(+), 7 deletions(-)
diff --git a/examples/Simple_Realistic_Railway_Generator.py b/examples/Simple_Realistic_Railway_Generator.py
index de3831a3..6df724f9 100644
--- a/examples/Simple_Realistic_Railway_Generator.py
+++ b/examples/Simple_Realistic_Railway_Generator.py
@@ -579,15 +579,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=100,
- height=100,
+ env = RailEnv(width=40+np.random.choice(100),
+ height=40+np.random.choice(100),
rail_generator=realistic_rail_generator(num_cities=2 + np.random.choice(10),
city_size=10 + np.random.choice(10),
- allowed_rotation_angles=[90],
- max_number_of_station_tracks=4,
- nbr_of_switches_per_station_track=2,
- connect_max_nbr_of_shortes_city=2,
- do_random_connect_stations=False,
+ allowed_rotation_angles=[-90,-45,0,45,90],
+ max_number_of_station_tracks=np.random.choice(4)+4,
+ nbr_of_switches_per_station_track=np.random.choice(4)+2,
+ connect_max_nbr_of_shortes_city=np.random.choice(4)+2,
+ do_random_connect_stations=np.random.choice(1)==0,
# Number of cities in map
seed=int(time.time()) # Random seed
),
--
GitLab