MazePlot wandb integration

[canrager]

I think the function should return figure and axes (i.e. use f, ax = plt.subplots() and ax.imshow()) this way the outputs can be passed to a wandb log call, or incorporated into another figure with subplots (ideally add a kwarg that directly accepts an axes object, in which case one doesn't need to create a new figure

Originally posted by @afspies in #108 (comment)

Alternatively: Add .to_wandb() function to MazePlot which returns the objects wandb log needs. @afspies which objects does wandb explicitly require for plotting?

[afspies]

wandb can be given a figure object (https://docs.wandb.ai/guides/track/log/plots#matplotlib-and-plotly-plots) or image array directly.

Indeed I think the correct behaviour is to have functions create figure objects unless an axis is passed via an optional kwarg.

If an axis is passed then this should be used directly (i.e. if one wants to have different subplots be filled in by plotting function calls)