FIX: stale propagation in artist with out fig/ax

James Evans · tacaswell · commit 955a2906c53f · 2015-08-11T15:53:50.000-04:00
Artists do not properly guard against being removed from figures or Axes
when tracking 'stale'. This will keep them from throwing an exception if
they become stale after being removed from an Axes or Figure.
diff --git a/lib/matplotlib/artist.py b/lib/matplotlib/artist.py
@@ -67,9 +67,9 @@ def draw_wrapper(artist, renderer, *args, **kwargs):
     draw_wrapper._supports_rasterization = True
     return draw_wrapper
 
-
-def _stale_axes_callback(self, value):
-    self.axes.stale = True
+def _stale_axes_callback(self):
+    if self.axes:
+        self.axes.stale = True
 
 
 class Artist(object):
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
@@ -52,7 +52,8 @@
 
 
 def _stale_figure_callback(self, val):
-    self.figure.stale = val
+    if self.figure:
+        self.figure.stale = val
 
 
 class AxesStack(Stack):
