From 7dceb3334342de86db2ccd89b88e77ecbde8c6da Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Mon, 16 Sep 2019 22:23:47 +0200
Subject: [PATCH] test
---
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 98a2e39e..f1649e78 100644
--- a/examples/Simple_Realistic_Railway_Generator.py
+++ b/examples/Simple_Realistic_Railway_Generator.py
@@ -468,19 +468,19 @@ for itrials in range(1000):
np.random.seed(0 * int(time.time()))
env = RailEnv(width=40 + np.random.choice(100),
height=40 + np.random.choice(100),
- rail_generator=realistic_rail_generator(num_cities=2000 + np.random.choice(10),
+ rail_generator=realistic_rail_generator(num_cities=2 + np.random.choice(10),
city_size=10 + np.random.choice(10),
allowed_rotation_angles=np.arange(-180, 180, 15),
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,
+ connect_max_nbr_of_shortes_city=2,
+ do_random_connect_stations=False and np.random.choice(1) == 0,
# Number of cities in map
seed=int(time.time()), # Random seed
print_out_info=False
),
schedule_generator=sparse_schedule_generator(),
- number_of_agents=1,
+ number_of_agents=100,
obs_builder_object=GlobalObsForRailEnv())
# reset to initialize agents_static
@@ -497,5 +497,4 @@ for itrials in range(1000):
"flatland_frame_{:04d}_{:04d}.png".format(itrials, 0)
))
- input()
env_renderer.close_window()
--
GitLab