Make Container._remove_method call correctly

jenshnielsen · Sep 21, 2012 · 843efd1 · 843efd1
1 parent 6b74c57
commit 843efd1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/matplotlib/axes.py b/lib/matplotlib/axes.py
@@ -1621,7 +1621,7 @@ def add_container(self, container):
         if not label:
             container.set_label('_container%d'%len(self.containers))
         self.containers.append(container)
-        container.set_remove_method(lambda h: self.containers.remove(container))
+        container.set_remove_method(lambda h: self.containers.remove(h))
         return container
 
 

diff --git a/lib/matplotlib/container.py b/lib/matplotlib/container.py
@@ -29,7 +29,7 @@ def remove(self):
             c.remove()
 
         if self._remove_method:
-            self._remove_method()
+            self._remove_method(self)
 
     def get_label(self):
         """