Adding a new method: FPNode#remove will remove the given child node.

Eric Naeseth · Eric Naeseth · commit 988339f8450b · 2009-03-10T03:25:33.000-05:00
diff --git a/fp_growth.py b/fp_growth.py
@@ -109,6 +109,15 @@ def search(self, item):
         except KeyError:
             return None
             
+    def remove(self, child):
+        try:
+            if self._children[child.item] is child:
+                del self._children[child.item]
+            else:
+                raise ValueError("that node is not a child of this node")
+        except KeyError:
+            raise ValueError("that node is not a child of this node")
+            
     def __contains__(self, item):
         return item in self._children
         
