Keep context of matplotlib plots in docstrings

Question

I know that I\'m able to keep the context of pieces of code in a .rst file with the option :context: close-figs.

I\'m trying to do the s