diff --git a/flatland/utils/rendertools.py b/flatland/utils/rendertools.py index 1a897d819c8579189d3752adb12436fce5845318..e5ca4ad58cadd04ba961377505ed89b4b1740797 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