From 5cdea1533d4c1b541a96b581b2b8f92f0dbd315a Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Sun, 6 Oct 2019 10:12:34 -0400
Subject: [PATCH] removinge trains with infeasible tasks
---
flatland/envs/schedule_generators.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/flatland/envs/schedule_generators.py b/flatland/envs/schedule_generators.py
index 07f10cec..4d783b8c 100644
--- a/flatland/envs/schedule_generators.py
+++ b/flatland/envs/schedule_generators.py
@@ -87,7 +87,8 @@ def sparse_schedule_generator(speed_ratio_map: Mapping[float, float] = None) ->
if not rail.check_path_exists(start[0], agent_orientation, target[0]):
agent_orientation = (agent_orientation + 2) % 4
if not (rail.check_path_exists(start[0], agent_orientation, target[0])):
- warnings.warn("Infeasible")
+ warnings.warn("Infeasible task: agent removed")
+ continue
agents_position.append((start[0][0], start[0][1]))
agents_target.append((target[0][0], target[0][1]))
--
GitLab