From 0310dd0743b10543e16bb367c3f1c5f9fed575a2 Mon Sep 17 00:00:00 2001
From: Erik Nygren <erik.nygren@sbb.ch>
Date: Mon, 15 Jul 2019 08:17:24 -0400
Subject: [PATCH] fixed previous bug in rendering the prediction. Now testing
if prediciton was actually generated before rendering it.
---
flatland/utils/rendertools.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/flatland/utils/rendertools.py b/flatland/utils/rendertools.py
index f3039e2d..fe1a3593 100644
--- a/flatland/utils/rendertools.py
+++ b/flatland/utils/rendertools.py
@@ -420,7 +420,7 @@ class RenderTool(object):
self.plot_agents(targets=True, selected_agent=selected_agent)
if show_observations:
self.render_observation(range(env.get_num_agents()), env.dev_obs_dict)
- if show_predictions:
+ if show_predictions and len(env.dev_pred_dict) > 0:
self.render_prediction(range(env.get_num_agents()), env.dev_pred_dict)
# Draw some textual information like fps
text_y = [-0.3, -0.6, -0.9]
@@ -554,7 +554,7 @@ class RenderTool(object):
if show_observations:
self.render_observation(range(env.get_num_agents()), env.dev_obs_dict)
- if show_predictions:
+ if show_predictions and len(env.dev_pred_dict) > 0:
self.render_prediction(range(env.get_num_agents()), env.dev_pred_dict)
if show:
self.gl.show()
--
GitLab