From f1379a0a3c3fb1f7a83599e278990209a89b2177 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Fri, 16 Aug 2019 21:15:42 +0200
Subject: [PATCH] bug fix in generator
---
flatland/envs/generators.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py
index aef00cdd..d7173fa3 100644
--- a/flatland/envs/generators.py
+++ b/flatland/envs/generators.py
@@ -753,7 +753,7 @@ def realistic_rail_generator(nr_start_goal=1, seed=0):
add_pos = (int((start[0] + goal[0]) / 2), int((start[1] + goal[1]) / 2))
agents_positions.append(add_pos)
- agents_directions.append(3)
+ agents_directions.append([1,3][nbr_track_loop % 2])
add_pos = (int((start[0] + goal[0]) / 2), int((2*start[1] + goal[1]) / 3))
agents_targets.append(add_pos)
--
GitLab