From bd86367606b028d9e83e2f3ee49dee6e2c9866cd Mon Sep 17 00:00:00 2001 From: Andrey Golovizin Date: Fri, 8 Aug 2014 13:11:29 +0200 Subject: [PATCH] Move _repr_png_ closer to the top. --- pixelocr/image.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/pixelocr/image.py b/pixelocr/image.py index d115486..6f2e407 100644 --- a/pixelocr/image.py +++ b/pixelocr/image.py @@ -7,13 +7,20 @@ from skimage.color import rgb2gray from .utils import cached_property -class Image(object): +class Image(IPythonImageMixin): + """Basic image class.""" + def __init__(self, data): self._data = data def __getitem__(self, key): return type(self)(self._data.__getitem__(key)) + def _repr_png_(self): + buf = BytesIO() + imsave(buf, self._data) + return buf.getvalue() + @classmethod def fromfile(cls, filename): return cls(imread(filename)) @@ -30,11 +37,6 @@ class Image(object): def T(self): return type(self)(self._data.swapaxes(0, 1)) - def _repr_png_(self): - buf = BytesIO() - imsave(buf, self._data) - return buf.getvalue() - @property def height(self): return self._data.shape[0]