What happened to heatmaps for plot_top_losses in fastai v2?

There used to be this very nice option heatmap=True in plot_top_losses to show heat maps for image classification results, but this feature seems to be gone in v2. What happened? What’s the way of getting heat maps for top losses in v2?