simplify axes clearing

Author: Sahil-Chhoker

mesa/visualization/backends/matplotlib_backend.py

@@ -47,8 +47,6 @@ def __init__(self, space_drawer):
         """
         super().__init__(space_drawer)
 
-        self._active_colorbars = []
-
     def initialize_canvas(self, ax=None):
         """Initialize the matplotlib canvas.
 
@@ -419,5 +417,4 @@ def draw_propertylayer(self, space, property_layers, propertylayer_portrayal):
                 sm = ScalarMappable(norm=norm, cmap=cmap)
                 sm.set_array([])
                 cbar = plt.colorbar(sm, ax=self.ax, label=layer_name)
-                self._active_colorbars.append(cbar)
         return self.ax, cbar

mesa/visualization/solara_viz.py

@@ -252,32 +252,12 @@ def SpaceRendererComponent(
 
     if renderer.backend == "matplotlib":
         # Clear the previous plotted data and agents
-        all_artists = [
-            renderer.canvas.lines[:],
-            renderer.canvas.collections[:],
-            renderer.canvas.patches[:],
-            renderer.canvas.images[:],
-            renderer.canvas.artists[:],
-        ]
-
-        # Remove duplicate colorbars from the canvas
-        for cbar in renderer.backend_renderer._active_colorbars:
-            cbar.remove()
-        renderer.backend_renderer._active_colorbars.clear()
-
-        # Chain them together into a single iterable
-        for artist in itertools.chain.from_iterable(all_artists):
-            artist.remove()
+        renderer.canvas.clear()
 
-        # Draw the space structure if specified
         if renderer.space_mesh:
             renderer.draw_structure()
-
-        # Draw agents if specified
         if renderer.agent_mesh:
             renderer.draw_agents()
-
-        # Draw property layers if specified
         if renderer.propertylayer_mesh:
             renderer.draw_propertylayer()