diff --git a/flatland/utils/graphics_layer.py b/flatland/utils/graphics_layer.py index de2e8618be143f287667a91be727a59ce4dfa4ec..b2dac4637d1560f9d1ac5ae485d8bf8bd5c2c97e 100644 --- a/flatland/utils/graphics_layer.py +++ b/flatland/utils/graphics_layer.py @@ -9,9 +9,6 @@ class GraphicsLayer(object): def open_window(self): pass - def is_raster(self): - return True - def plot(self, *args, **kwargs): pass diff --git a/flatland/utils/graphics_pil.py b/flatland/utils/graphics_pil.py index f8920afd8e39836e2be72aa32a3ebb4d926560f7..cf91cd67c1f29554b42eb080aed066700899f489 100644 --- a/flatland/utils/graphics_pil.py +++ b/flatland/utils/graphics_pil.py @@ -274,9 +274,6 @@ class PILSVG(PILGL): self.load_rail() self.load_agent() - def is_raster(self): - return False - def process_events(self): time.sleep(0.001) diff --git a/flatland/utils/rendertools.py b/flatland/utils/rendertools.py index 4aca5c386790784bd4f7dde1d387d2f06ac2e0f0..d361957294da6d6de1532ae64143909d6a789d97 100644 --- a/flatland/utils/rendertools.py +++ b/flatland/utils/rendertools.py @@ -403,7 +403,7 @@ class RenderTool(object): """ Draw the environment using the GraphicsLayer this RenderTool was created with. (Use show=False from a Jupyter notebook with %matplotlib inline) """ - if not self.gl.is_raster(): + if type(self.gl) is PILSVG: self.render_env_svg(show=show, show_observations=show_observations, show_predictions=show_predictions,