From 0213fa049e2ad62007581ca320f82b05e196c956 Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Tue, 1 Oct 2019 13:37:27 -0400
Subject: [PATCH] agents randomly facing any outgoing direciton of city
---
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 30f6b57e..1ac3663e 100644
--- a/flatland/envs/schedule_generators.py
+++ b/flatland/envs/schedule_generators.py
@@ -74,13 +74,14 @@ def sparse_schedule_generator(speed_ratio_map: Mapping[float, float] = None) ->
# Set target for agent
start_city = agent_start_targets_nodes[agent_idx][0]
target_city = agent_start_targets_nodes[agent_idx][0]
- agent_orientation = agent_start_targets_nodes[agent_idx][2]
+ 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]
target_station_idx = np.random.randint(len(train_stations[target_city]))
target = train_stations[target_city][target_station_idx]
agents_position.append((start[0][0], start[0][1]))
agents_target.append((target[0][0], target[0][1]))
+
agents_direction.append(agent_orientation)
# Orient the agent correctly
--
GitLab