From 52ae5db7f17b1e577200005bd9ba206f69b0a312 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Tue, 17 Sep 2019 16:29:22 +0200
Subject: [PATCH] PEP-8 fix
---
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 04e5952f..3817c50d 100644
--- a/examples/Simple_Realistic_Railway_Generator.py
+++ b/examples/Simple_Realistic_Railway_Generator.py
@@ -151,9 +151,9 @@ def realistic_rail_generator(num_cities=5,
track = datas[0]
if len(track) > 0:
if k % 2 == 0:
- x = int(np.random.choice(int(len(track)/2))+1)
+ x = int(np.random.choice(int(len(track) / 2)) + 1)
else:
- x = len(track) - int(np.random.choice(int(len(track)/2))+1)
+ x = len(track) - int(np.random.choice(int(len(track) / 2)) + 1)
start_node = track[x]
for i in np.arange(1, len(datas)):
track = datas[i]
@@ -174,7 +174,7 @@ def realistic_rail_generator(num_cities=5,
print("create_switches_at_stations : connect_rail -> no path found")
start_node = datas[i][0]
end_node = datas[i - 1][0]
- connection = connect_rail(rail_trans, grid_map, start_node, end_node)
+ connect_rail(rail_trans, grid_map, start_node, end_node)
nodes_added.append(start_node)
nodes_added.append(end_node)
@@ -484,16 +484,16 @@ if os.path.exists("./../render_output/"):
np.random.seed(itrials)
env = RailEnv(width=40 + np.random.choice(100),
height=40 + np.random.choice(100),
- rail_generator=realistic_rail_generator(num_cities=1000,
+ rail_generator=realistic_rail_generator(num_cities=5 + np.random.choice(10),
city_size=10 + np.random.choice(10),
- allowed_rotation_angles=np.arange(-180, 180, 90),
+ allowed_rotation_angles=np.arange(-180, 180, 30),
max_number_of_station_tracks=1 + np.random.choice(4),
- nbr_of_switches_per_station_track=2,
+ nbr_of_switches_per_station_track=2 + np.random.choice(2),
connect_max_nbr_of_shortes_city=2 + np.random.choice(4),
- do_random_connect_stations=False,
+ do_random_connect_stations=itrials % 2 == 0,
# Number of cities in map
seed=itrials, # Random seed
- print_out_info=True
+ print_out_info=False
),
schedule_generator=sparse_schedule_generator(),
number_of_agents=0,
--
GitLab