Solara: Should API for custom space drawer be simplified further?

[rht]

Currently, this is how you would specify a custom space drawer: #1769 (comment)

def space_drawer(viz):
    space_fig = Figure()
    space_ax = space_fig.subplots()
    graph = viz.model.grid.G
    pos = nx.circular_layout(graph, seed=0)
    nx.draw(
        graph,
        ax=space_ax,
        pos=pos,
        **agent_portrayal(graph),
    )
    space_ax.set_axis_off()
    solara.FigureMatplotlib(space_fig, dependencies=[viz.model, viz.df])

I'm tempted to simplify it to

def space_drawer(viz, fig, ax):
    graph = viz.model.grid.G
    pos = nx.circular_layout(graph, seed=0)
    nx.draw(
        graph,
        ax=ax,
        pos=pos,
        **agent_portrayal(graph),
    )

But the drawback is that it is hardcoded to be a Matplotlib visualization. Some people might want to use other Solara-supported visualization, such as Plotly.

[rht]

I think this is a good solution:

from mesa.experimental import prepare_matplotlib_space

@prepare_matplotlib_space
def space_drawer(viz, fig, ax):
    graph = viz.model.grid.G
    pos = nx.circular_layout(graph, seed=0)
    nx.draw(
        graph,
        ax=ax,
        pos=pos,
        **agent_portrayal(graph),
    )

[rht]

Implemented in #1771.