It would be useful if the learner2D.plot() function had the colormap as an input, instead of choosing viridis as default.
I am not sure whether this is possible with the current version of Adaptive, but I did not find it implemented. I made it work by writing a slightly different version of the function that takes the cmap as an input.