FIX: make image.set_clim mark the figure as stale

tacaswell · tacaswell · commit 818624359660 · 2017-08-29T22:18:47.000-04:00
Push the stale flag down into the ScalarMappable so that it gets
applied to all ScalarMappable sub-classes (ex Collections).
diff --git a/lib/matplotlib/cm.py b/lib/matplotlib/cm.py
@@ -311,6 +311,7 @@ def set_clim(self, vmin=None, vmax=None):
         if vmax is not None:
             self.norm.vmax = vmax
         self.changed()
+        self.stale = True
 
     def set_cmap(self, cmap):
         """
@@ -321,13 +322,15 @@ def set_cmap(self, cmap):
         cmap = get_cmap(cmap)
         self.cmap = cmap
         self.changed()
+        self.stale = True
 
     def set_norm(self, norm):
         'set the normalization instance'
         if norm is None:
             norm = colors.Normalize()
         self.norm = norm
         self.changed()
+        self.stale = True
 
     def autoscale(self):
         """
diff --git a/lib/matplotlib/image.py b/lib/matplotlib/image.py
@@ -192,14 +192,6 @@ def _interpdr(self):
     def iterpnames(self):
         return interpolations_names
 
-    def set_cmap(self, cmap):
-        super(_ImageBase, self).set_cmap(cmap)
-        self.stale = True
-
-    def set_norm(self, norm):
-        super(_ImageBase, self).set_norm(norm)
-        self.stale = True
-
     def __str__(self):
         return "AxesImage(%g,%g;%gx%g)" % tuple(self.axes.bbox.bounds)
 
