Merge pull request #19117 from QuLogic/rename-var

timhoffm · web-flow · commit cf4f8e10ee1d · 2020-12-15T09:00:12.000+01:00
Rename a confusing variable.
diff --git a/lib/matplotlib/backends/backend_webagg_core.py b/lib/matplotlib/backends/backend_webagg_core.py
@@ -196,14 +196,15 @@ def get_diff_image(self):
                 diff = buff != self._last_buff
                 output = np.where(diff, buff, 0)
 
-            buf = BytesIO()
-            data = output.view(dtype=np.uint8).reshape((*output.shape, 4))
-            Image.fromarray(data).save(buf, format="png")
-            # store the current buffer so we can compute the next diff
+            # Store the current buffer so we can compute the next diff.
             np.copyto(self._last_buff, buff)
             self._force_full = False
             self._png_is_old = False
-            return buf.getvalue()
+
+            data = output.view(dtype=np.uint8).reshape((*output.shape, 4))
+            with BytesIO() as png:
+                Image.fromarray(data).save(png, format="png")
+                return png.getvalue()
 
     def get_renderer(self, cleared=None):
         # Mirrors super.get_renderer, but caches the old one so that we can do
