Replace pygtrie by a simplistic implementation

Author: dbarbier

mlxtend/frequent_patterns/apriori.py

@@ -6,10 +6,34 @@
 
 import numpy as np
 import pandas as pd
-import pygtrie
 from ..frequent_patterns import fpcommon as fpc
 
 
+class _FixedLengthTrie:
+    __slots__ = ("root")
+
+    def __init__(self, combinations):
+        self.root = dict()
+        for combination in combinations:
+            current = self.root
+            for item in combination:
+                try:
+                    current = current[item]
+                except KeyError:
+                    next_node = dict()
+                    current[item] = next_node
+                    current = next_node
+
+    def __contains__(self, combination):
+        current = self.root
+        try:
+            for item in combination:
+                current = current[item]
+            return True
+        except KeyError:
+            return False
+
+
 def generate_new_combinations(old_combinations):
     """
     Generator of all combinations based on the last state of Apriori algorithm
@@ -44,7 +68,7 @@ def generate_new_combinations(old_combinations):
     """
 
     length = len(old_combinations)
-    trie = pygtrie.Trie(list(zip(old_combinations, [1]*length)))
+    trie = _FixedLengthTrie(old_combinations)
     for i, old_combination in enumerate(old_combinations):
         *head_i, _ = old_combination
         j = i + 1