From b1f5d59f56f5a4f1020f3a95efe483f45b0576db Mon Sep 17 00:00:00 2001
From: u229589 <christian.baumberger@sbb.ch>
Date: Wed, 25 Sep 2019 09:44:55 +0200
Subject: [PATCH] remove assert in production code
---
flatland/envs/rail_env_utils.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/flatland/envs/rail_env_utils.py b/flatland/envs/rail_env_utils.py
index c12a3882..69cfce76 100644
--- a/flatland/envs/rail_env_utils.py
+++ b/flatland/envs/rail_env_utils.py
@@ -24,7 +24,9 @@ def load_flatland_environment_from_file(file_name, load_from_package=None, obs_b
def visualize_distance_map(distance_map: DistanceMap, agent_handle: int = 0):
- assert agent_handle < distance_map.get().shape[0]
+ if agent_handle >= distance_map.get().shape[0]:
+ print("Error: agent_handle cannot be larger than actual number of agents")
+ return
# take min value of all 4 directions
min_distance_map = np.min(distance_map.get(), axis=3)
plt.imshow(min_distance_map[agent_handle][:][:])
--
GitLab