Made the code more concise by avoiding try-catch and using lambda.

onatt · onatt · commit 03d41e891a56 · 2019-02-23T11:19:40.000+01:00
diff --git a/lib/matplotlib/animation.py b/lib/matplotlib/animation.py
@@ -1222,14 +1222,11 @@ def _setup_blit(self):
         # axes
         self._blit_cache = dict()
         self._drawn_artists = []
-        def remove_from_cache(ax):
-            try:
-                del self._blit_cache[ax]
-            except KeyError:
-                pass
         for ax in self._fig.axes:
-            ax.callbacks.connect('xlim_changed', remove_from_cache)
-            ax.callbacks.connect('ylim_changed', remove_from_cache)
+            ax.callbacks.connect('xlim_changed',
+                                 lambda ax: self._blit_cache.pop(ax, None))
+            ax.callbacks.connect('ylim_changed',
+                                 lambda ax: self._blit_cache.pop(ax, None))
         self._resize_id = self._fig.canvas.mpl_connect('resize_event',
                                                        self._handle_resize)
         self._post_draw(None, self._blit)
