add keyerro in get_colormap_py