we could just put the import inside the plot function
I didn't see `have_matplotlib` used anywhere in this module