Clean up outside the loop

vallsv · vallsv · commit e7d6f1284ee2 · 2021-02-15T13:56:13.000+01:00
diff --git a/lib/matplotlib/cbook/__init__.py b/lib/matplotlib/cbook/__init__.py
@@ -219,11 +219,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):
         """
