Eu quero usar o mapa de cores "viridis" (http://bids.github.io/colormap/) e ainda não atualizarei a versão de desenvolvimento 1.5. Assim, eu baixeicolormaps.py dehttps://github.com/BIDS/colormap. Infelizmente, não sou capaz de fazê-lo funcionar. Isto é o que eu faço:

import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import numpy as np

import colormaps as cmaps

img=mpimg.imread('stinkbug.png')
lum_img = np.flipud(img[:,:,0])

plt.set_cmap(cmaps.viridis)
imgplot = plt.pcolormesh(lum_img)

Isso me dá umaValueError, o retorno que termina com,

ValueError: o Colormap viridis não é reconhecido. Os valores possíveis são: espectral, verão, frio, ...

(E depois a lista completa dos mapas de cores originalmente instalados.)

Alguma idéia de como corrigir esse problema?