sugarscape G1MT: Simplify custom space drawer

Author: rht

examples/sugarscape_g1mt/app.py

@@ -1,13 +1,14 @@
 import numpy as np
 import solara
 from matplotlib.figure import Figure
-from mesa.experimental import JupyterViz
+from mesa.experimental import JupyterViz, prepare_matplotlib_space
 from sugarscape_g1mt.model import SugarscapeG1mt
 from sugarscape_g1mt.resource_agents import Sugar
 from sugarscape_g1mt.trader_agents import Trader
 
 
-def space_drawer(viz):
+@prepare_matplotlib_space
+def space_drawer(viz, fig, ax):
     def portray(g):
         layers = {
             "sugar": [[np.nan for j in range(g.height)] for i in range(g.width)],
@@ -29,8 +30,6 @@ def portray(g):
                         layers["spice"][i][j] = value
         return layers
 
-    fig = Figure()
-    ax = fig.subplots()
     out = portray(viz.model.grid)
     # Sugar
     # Important note: imshow by default draws from upper left. You have to
@@ -41,8 +40,6 @@ def portray(g):
     ax.imshow(out["spice"], cmap="winter", origin="lower")
     # Trader
     ax.scatter(**out["trader"])
-    ax.set_axis_off()
-    solara.FigureMatplotlib(fig, dependencies=[viz.model, viz.df])
 
 
 model_params = {