DOC: update outdated plot_heatmap docstring