Clean up outside the loop

vallsv · vallsv · commit cfe1f3efdc4a · 2021-02-08T14:58:09.000+01:00
diff --git a/lib/matplotlib/cbook/__init__.py b/lib/matplotlib/cbook/__init__.py
@@ -220,11 +220,15 @@ def _remove_proxy(self, proxy, *, _is_finalizing=sys.is_finalizing):
         for signal, proxies in list(self._func_cid_map.items()):
             try:
                 del self.callbacks[signal][proxies[proxy]]
+                break
             except KeyError:
                 pass
-            if len(self.callbacks[signal]) == 0:
-                del self.callbacks[signal]
-                del self._func_cid_map[signal]
+        else:
+            # Not found
+            return
+        if len(self.callbacks[signal]) == 0:
+            del self.callbacks[signal]
+            del self._func_cid_map[signal]
 
     def disconnect(self, cid):
         """
