From 32c11d525de0e8bd0a95ec2f9260ee4e540f69d4 Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Wed, 25 Sep 2019 18:22:15 -0400
Subject: [PATCH] minor fixes to code to get tests running again.
---
flatland/envs/rail_generators.py | 2 +-
flatland/envs/schedule_generators.py | 1 +
2 files changed, 2 insertions(+), 1 deletion(-)
diff --git a/flatland/envs/rail_generators.py b/flatland/envs/rail_generators.py
index de780402..147b1bd1 100644
--- a/flatland/envs/rail_generators.py
+++ b/flatland/envs/rail_generators.py
@@ -781,7 +781,7 @@ def sparse_rail_generator(num_cities=5, num_intersections=4, num_trainstations=2
node_available_target.append(len(train_stations[node_idx]))
# Assign agents to slots
- for agent_idx in range(nb_nodes):
+ for agent_idx in range(num_agents):
avail_start_nodes = [idx for idx, val in enumerate(node_available_start) if val > 0]
avail_target_nodes = [idx for idx, val in enumerate(node_available_target) if val > 0]
start_node = np.random.choice(avail_start_nodes)
diff --git a/flatland/envs/schedule_generators.py b/flatland/envs/schedule_generators.py
index b3576a2b..8c183490 100644
--- a/flatland/envs/schedule_generators.py
+++ b/flatland/envs/schedule_generators.py
@@ -68,6 +68,7 @@ def sparse_schedule_generator(speed_ratio_map: Mapping[float, float] = None) ->
agents_position = []
agents_target = []
agents_direction = []
+
for agent_idx in range(num_agents):
# Set target for agent
current_target_node = agent_start_targets_nodes[agent_idx][1]
--
GitLab