diff options
| -rw-r--r-- | native/src/unigram_dictionary.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/native/src/unigram_dictionary.cpp b/native/src/unigram_dictionary.cpp index 9f8f04e50..b95da99a3 100644 --- a/native/src/unigram_dictionary.cpp +++ b/native/src/unigram_dictionary.cpp @@ -808,7 +808,8 @@ inline bool UnigramDictionary::processCurrentNode(const int initialPos, const in } } // Optimization: Prune out words that are too long compared to how much was typed. - if (correctionState->getOutputIndex() >= maxDepth || diffs > mMaxEditDistance) { + if (isTerminal + && (correctionState->getOutputIndex() >= maxDepth || diffs > mMaxEditDistance)) { // We are giving up parsing this node and its children. Skip the rest of the node, // output the sibling position, and return that we don't want to traverse children. if (!isLastChar) { |