matplotlib
diff --git a/‎examples/pylab_examples/agg_buffer.py
Copy file name to clipboardExpand all lines: examples/pylab_examples/agg_buffer.py
+6-1Lines changed: 6 additions & 1 deletion b/‎examples/pylab_examples/agg_buffer.py
Copy file name to clipboardExpand all lines: examples/pylab_examples/agg_buffer.py
+6-1Lines changed: 6 additions & 1 deletion
@@ -4,6 +4,8 @@
 convert it to an array and pass it to Pillow for rendering.
 """
 
+import numpy as np
+
 import matplotlib.pyplot as plt
 from matplotlib.backends.backend_agg import FigureCanvasAgg
 
@@ -28,7 +30,10 @@
 X = np.fromstring(s, np.uint8)
 X.shape = h, w, 3
 
-im = Image.fromstring("RGB", (w, h), s)
+try:
+    im = Image.fromstring("RGB", (w, h), s)
+except Exception:
+    im = Image.frombytes("RGB", (w, h), s)
 
 # Uncomment this line to display the image using ImageMagick's
 # `display` tool.