import cmocean
cmocean.plots.wrap_viscm(cmocean.cm.haline, saveplot=False)