From 510866a39bf417a501d1439ff5163802554b0ae7 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Wed, 18 Sep 2019 20:45:21 +0200
Subject: [PATCH] bug fix
---
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 e81da924..9b90e7c1 100644
--- a/flatland/envs/schedule_generators.py
+++ b/flatland/envs/schedule_generators.py
@@ -103,7 +103,7 @@ def sparse_schedule_generator(speed_ratio_map: Mapping[float, float] = None) ->
transitions = rail.get_transitions(start[0], start[1], orientation)
if any(transitions) > 0:
agents_direction.append(orientation)
- continue
+ break
if speed_ratio_map:
speeds = speed_initialization_helper(num_agents, speed_ratio_map)
--
GitLab