From 69b8db28a54790dad8713eeaed9e597f72c1473a Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Tue, 1 Oct 2019 13:38:18 -0400
Subject: [PATCH] fixed target distribution bug
---
flatland/envs/schedule_generators.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/flatland/envs/schedule_generators.py b/flatland/envs/schedule_generators.py
index 1ac3663e..2b572bfe 100644
--- a/flatland/envs/schedule_generators.py
+++ b/flatland/envs/schedule_generators.py
@@ -73,7 +73,7 @@ def sparse_schedule_generator(speed_ratio_map: Mapping[float, float] = None) ->
for agent_idx in range(num_agents):
# Set target for agent
start_city = agent_start_targets_nodes[agent_idx][0]
- target_city = agent_start_targets_nodes[agent_idx][0]
+ target_city = agent_start_targets_nodes[agent_idx][1]
agent_orientation = (agent_start_targets_nodes[agent_idx][2] + np.random.choice([0, 2])) % 4
start_city_idx = np.random.randint(len(train_stations[start_city]))
start = train_stations[start_city][start_city_idx]
--
GitLab