Update plot.py

fix matplotlib.pyplot error

Author: sokrypton

colabdesign/shared/plot.py

@@ -150,6 +150,7 @@ def rescale(a, amin=None, amax=None):
 
   colors[:,:3] = colors[:,:3] + (1 - colors[:,:3]) * (0.50 * z + 0.50 * tint_mask) / 3
   colors[:,:3] = colors[:,:3] * (0.20 + 0.25 * z + 0.55 * shadow_mask)
+  colors = np.clip(colors,0,1)
 
   set_lim = False
   if ax is None:
@@ -333,4 +334,4 @@ def nankabsch(a,b,**kwargs):
   # make animation!
   ani = animation.ArtistAnimation(fig, ims, blit=True, interval=interval)
   plt.close()
-  return ani.to_html5_video()
+  return ani.to_html5_video()