diff --git a/flatland/utils/rendertools.py b/flatland/utils/rendertools.py index af5f68ff0334c29576ad0e72ef55c4ccf571e21a..75f61507caa5dd76d6936c63cdf1fc0824ded0c2 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. @@ -418,7 +418,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