minor tweak to pyplot.savefig to ensure that if transparent=True

tacaswell · tacaswell · commit 99c8278b0e3f · 2013-09-07T10:53:14.000-05:00
is passed to `savefig`, the patch colors are restored on the screen
after saving.
diff --git a/lib/matplotlib/pyplot.py b/lib/matplotlib/pyplot.py
@@ -563,7 +563,9 @@ def draw():
 @docstring.copy_dedent(Figure.savefig)
 def savefig(*args, **kwargs):
     fig = gcf()
-    return fig.savefig(*args, **kwargs)
+    res = fig.savefig(*args, **kwargs)
+    draw()   # need this if 'transparent=True' to reset colors
+    return res
 
 
 @docstring.copy_dedent(Figure.ginput)
