JupyterViz: Convert make_plot to Solara component

Author: rht

mesa/experimental/components/matplotlib.py

@@ -112,7 +112,8 @@ def portray(space):
     space_ax.scatter(**portray(space))
 
 
-def make_plot(model, measure):
+@solara.component
+def PlotMatplotlib(model, measure, dependencies: Optional[list[any]] = None):
     fig = Figure()
     ax = fig.subplots()
     df = model.datacollector.get_model_vars_dataframe()
@@ -129,4 +130,4 @@ def make_plot(model, measure):
         fig.legend()
     # Set integer x axis
     ax.xaxis.set_major_locator(MaxNLocator(integer=True))
-    solara.FigureMatplotlib(fig)
+    solara.FigureMatplotlib(fig, dependencies=dependencies)

mesa/experimental/jupyter_viz.py

@@ -37,7 +37,9 @@ def Card(
                 # Is a custom object
                 measure(model)
             else:
-                components_matplotlib.make_plot(model, measure)
+                components_matplotlib.PlotMatplotlib(
+                    model, measure, dependencies=[current_step.value]
+                )
     return main
 
 
@@ -116,13 +118,14 @@ def render_in_jupyter():
             # otherwise, do nothing (do not draw space)
 
             # 5. Plots
-
             for measure in measures:
                 if callable(measure):
                     # Is a custom object
                     measure(model)
                 else:
-                    components_matplotlib.make_plot(model, measure)
+                    components_matplotlib.PlotMatplotlib(
+                        model, measure, dependencies=[current_step.value]
+                    )
 
     def render_in_browser():
         # if space drawer is disabled, do not include it