From 2e47c1393bd60f6f352c31ea649eb15cc8273814 Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Wed, 2 Oct 2019 12:03:14 -0400
Subject: [PATCH] squashed bug where agent was placed in target city
---
examples/flatland_2_0_example.py | 12 ++++++------
flatland/envs/schedule_generators.py | 2 +-
2 files changed, 7 insertions(+), 7 deletions(-)
diff --git a/examples/flatland_2_0_example.py b/examples/flatland_2_0_example.py
index d6e9545d..cf8f635d 100644
--- a/examples/flatland_2_0_example.py
+++ b/examples/flatland_2_0_example.py
@@ -28,16 +28,16 @@ speed_ration_map = {1.: 0.25, # Fast passenger train
1. / 3.: 0.25, # Slow commuter train
1. / 4.: 0.25} # Slow freight train
-env = RailEnv(width=50,
- height=50,
- rail_generator=sparse_rail_generator(num_cities=10, # Number of cities in map (where train stations are)
- seed=10, # Random seed
+env = RailEnv(width=75,
+ height=75,
+ rail_generator=sparse_rail_generator(num_cities=15, # Number of cities in map (where train stations are)
+ seed=1, # Random seed
grid_mode=False,
max_inter_city_rails=2,
- max_tracks_in_city=4,
+ max_tracks_in_city=8,
),
schedule_generator=sparse_schedule_generator(speed_ration_map),
- number_of_agents=50,
+ number_of_agents=200,
stochastic_data=stochastic_data, # Malfunction data generator
obs_builder_object=GlobalObsForRailEnv(),
remove_agents_at_target=True
diff --git a/flatland/envs/schedule_generators.py b/flatland/envs/schedule_generators.py
index b4d60713..c7be5f57 100644
--- a/flatland/envs/schedule_generators.py
+++ b/flatland/envs/schedule_generators.py
@@ -80,7 +80,7 @@ def sparse_schedule_generator(speed_ratio_map: Mapping[float, float] = None) ->
while start[1] % 2 != 0:
start = random.choice(train_stations[start_city])
while target[1] % 2 != 1:
- target = random.choice(train_stations[start_city])
+ target = random.choice(train_stations[target_city])
agent_orientation = (agent_start_targets_nodes[agent_idx][2] + 2 * start[1]) % 4
if not rail.check_path_exists(start[0], agent_orientation, target[0]):
--
GitLab