From 46a1ec87594f71562e8f60680d06b3010e4e32b7 Mon Sep 17 00:00:00 2001
From: MLErik <baerenjesus@gmail.com>
Date: Fri, 4 Oct 2019 14:46:28 -0400
Subject: [PATCH] revert index to get tests green
---
flatland/envs/rail_generators.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/flatland/envs/rail_generators.py b/flatland/envs/rail_generators.py
index 6532d980..9c483537 100644
--- a/flatland/envs/rail_generators.py
+++ b/flatland/envs/rail_generators.py
@@ -700,7 +700,7 @@ def sparse_rail_generator(max_num_cities: int = 5, grid_mode: bool = False, max_
tmp_coordinates = (
city_position[0] + connection_slots[connection_idx], city_position[1] - city_radius)
connection_points_coordinates_inner[direction].append(tmp_coordinates)
- if connection_idx in range(start_idx, start_idx + number_of_out_rails):
+ if connection_idx in range(start_idx, start_idx + number_of_out_rails + 1):
connection_points_coordinates_outer[direction].append(tmp_coordinates)
inner_connection_points.append(connection_points_coordinates_inner)
--
GitLab