From 028d22be0ac185b788e4350868115bc8b5e478f7 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Fri, 16 Aug 2019 14:20:34 +0200
Subject: [PATCH] agent and target placement
---
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 4ada5465..d63e161a 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(np.random.choice([3, 1]))
+ agents_directions.append(3)
add_pos = (int((start[0] + goal[0]) / 2), int((2*start[1] + goal[1]) / 3))
agents_targets.append(add_pos)
--
GitLab