From 7aa2542a666a2ce75596960741fe583ebfd89144 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Thu, 12 Sep 2019 16:19:18 +0200
Subject: [PATCH] generator
---
examples/Simple_Realistic_Railway_Generator.py | 9 ++++-----
1 file changed, 4 insertions(+), 5 deletions(-)
diff --git a/examples/Simple_Realistic_Railway_Generator.py b/examples/Simple_Realistic_Railway_Generator.py
index 82a88b5b..859c34a4 100644
--- a/examples/Simple_Realistic_Railway_Generator.py
+++ b/examples/Simple_Realistic_Railway_Generator.py
@@ -321,11 +321,10 @@ def realistic_rail_generator(num_cities=5,
s_nodes = org_s_nodes.copy()
e_nodes = org_e_nodes.copy()
-
for city_loop in range(len(s_nodes)):
+ old_cl = -1
for k in range(inter_max_number_of_connecting_tracks):
sns = s_nodes[city_loop]
- old_cl = -1
min_distance = np.inf
end_node = None
start_node = None
@@ -526,9 +525,9 @@ for itrials in range(100):
np.random.seed(int(time.time()))
env = RailEnv(width=100, # 20+np.random.choice(100),
height=100, # 20+np.random.choice(100),
- rail_generator=realistic_rail_generator(num_cities=np.random.choice(10)+2,
- city_size=np.random.choice(10)+10,
- allowed_rotation_angles=np.arange(0,360,45),
+ rail_generator=realistic_rail_generator(num_cities=np.random.choice(10) + 2,
+ city_size=np.random.choice(10) + 10,
+ allowed_rotation_angles=np.arange(0, 360, 45),
max_number_of_station_tracks=4,
nbr_of_switches_per_station_track=2,
max_number_of_connecting_tracks=4,
--
GitLab