From 65afa67a6beaf3a0fb8d45587e1c7ded37f248de Mon Sep 17 00:00:00 2001
From: hagrid67 <jdhwatson@gmail.com>
Date: Tue, 9 Apr 2019 17:25:38 +0100
Subject: [PATCH] adding boolean show arg to renderEnv - for use in / out of
notebooks
---
flatland/utils/rendertools.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/flatland/utils/rendertools.py b/flatland/utils/rendertools.py
index 1a897d8..e5ca4ad 100644
--- a/flatland/utils/rendertools.py
+++ b/flatland/utils/rendertools.py
@@ -289,7 +289,7 @@ class RenderTool(object):
visit = visit.prev
xyPrev = xy
- def renderEnv(self):
+ def renderEnv(self, show=False):
"""
Draw the environment using matplotlib.
Draw into the figure if provided.
@@ -419,7 +419,8 @@ class RenderTool(object):
plt.xlim([0, env.width * cell_size])
plt.ylim([-env.height * cell_size, 0])
- plt.show()
+ if show:
+ plt.show()
def _draw_square(self, center, size, color):
x0 = center[0]-size/2
--
GitLab