Fix: save correct callback ids.

tonysyu · tonysyu · commit 6baa65a592a4 · 2012-02-28T13:30:51.000-05:00
Also, remove unused import.
diff --git a/lib/matplotlib/widgets.py b/lib/matplotlib/widgets.py
@@ -10,7 +10,6 @@
 """
 
 from __future__ import print_function
-import functools
 import numpy as np
 
 from mlab import dist
@@ -82,8 +81,8 @@ def __init__(self, ax):
         self.active = True
 
     def connect_event(self, event, callback):
-        self.canvas.mpl_connect(event, callback)
-        self.cids.append(callback)
+        cid = self.canvas.mpl_connect(event, callback)
+        self.cids.append(cid)
 
     def disconnect_events(self):
         for c in self.cids:
