Merge pull request #5844 from radarhere/numpy_array

python-pillow · Nov 21, 2021 · 059503b · 059503b
2 parents e1eefe4 + 83c42fc
commit 059503b
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/PIL/Image.py b/src/PIL/Image.py
@@ -656,6 +656,10 @@ def _repr_png_(self):
             raise ValueError("Could not save to PNG for display") from e
         return b.getvalue()
 
+    class _ArrayData:
+        def __init__(self, new):
+            self.__array_interface__ = new
+
     def __array__(self, dtype=None):
         # numpy array interface support
         import numpy as np
@@ -672,10 +676,7 @@ def __array__(self, dtype=None):
         else:
             new["data"] = self.tobytes()
 
-        class ArrayData:
-            __array_interface__ = new
-
-        return np.array(ArrayData(), dtype)
+        return np.array(self._ArrayData(new), dtype)
 
     def __getstate__(self):
         return [self.info, self.mode, self.size, self.getpalette(), self.tobytes()]