Check if suffix is zero state earlier. Improves setup performance sig…

…nificantly. Fixes #8
abusix · Aug 19, 2020 · 06570c9 · 06570c9
1 parent 7098bbb
commit 06570c9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/ahocorapy/keywordtree.py b/src/ahocorapy/keywordtree.py
@@ -162,11 +162,12 @@ def search_lss(self, state):
                 else:
                     traversed = traversed.longest_strict_suffix
             suffix = state.longest_strict_suffix
+            if suffix == self._zero_state:
+                return
             if suffix.longest_strict_suffix is None:
                 self.search_lss(suffix)
             for symbol, next_state in suffix.transitions.items():
-                if (symbol not in state.transitions and
-                        suffix != self._zero_state):
+                if symbol not in state.transitions:
                     state.transitions[symbol] = next_state
 
     def __str__(self):