From 27058ffee86d057e71749e6f6ccb0edba43cc891 Mon Sep 17 00:00:00 2001 From: caspervdw Date: Thu, 5 Nov 2015 21:34:43 +0100 Subject: [PATCH] FIX Image import in plot_to_frame --- pims/display.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/pims/display.py b/pims/display.py index 105248a4..e9403806 100644 --- a/pims/display.py +++ b/pims/display.py @@ -391,7 +391,11 @@ def plot_to_frame(fig, dpi, **imsave_kwargs): pims.Frame object containing RGBA values (dtype uint8) """ if mpl is None: - raise ImportError('Module matplotlib not found') + raise ImportError("This feature requires matplotlib.") + try: + from PIL import Image + except ImportError: + raise ImportError("This feature requires PIL/Pillow.") from pims import Frame buffer = six.BytesIO() if isinstance(fig, mpl.axes.Axes): @@ -419,7 +423,7 @@ def plots_to_frame(figures, width=512, close_fig=False, **imsave_kwargs): pims.Frame object containing a stack of RGBA values (dtype uint8) """ if mpl is None: - raise ImportError('Module matplotlib not found') + raise ImportError("This feature requires matplotlib.") from pims import Frame if 'dpi' in imsave_kwargs or 'format' in imsave_kwargs: raise ValueError('Do not specify dpi or format imsave kwargs.')