FIX: use toplevel function so pickle works

tacaswell · tacaswell · commit 1d66b42ecbd0 · 2015-06-08T16:43:03.000-04:00
diff --git a/lib/matplotlib/pyplot.py b/lib/matplotlib/pyplot.py
@@ -530,15 +530,27 @@ def make_active(event):
 
         _pylab_helpers.Gcf.set_active(figManager)
         figManager.canvas.figure.number = num
+
         if _INSTALL_FIG_OBSERVER:
-            def auto_draw(fig):
-                if fig.stale and matplotlib.is_interactive():
-                    fig.canvas.draw_idle()
-            figManager.canvas.figure.add_callback(auto_draw)
+            figManager.canvas.figure.add_callback(_auto_draw_if_interactive)
 
     return figManager.canvas.figure
 
 
+def _auto_draw_if_interactive(fig):
+    """
+    This is an internal helper function for making sure that auto-redrawing
+    works as intended in the plain python repl.
+
+    Parameters
+    ----------
+    fig : Figure
+        A figure object which is assumed to be associated with a canvas
+    """
+    if fig.stale and matplotlib.is_interactive():
+        fig.canvas.draw_idle()
+
+
 def gcf():
     "Get a reference to the current figure."
 
@@ -548,6 +560,7 @@ def gcf():
     else:
         return figure()
 
+
 def fignum_exists(num):
     return _pylab_helpers.Gcf.has_fignum(num) or num in get_figlabels()
 
