Fix a regression that mutScore is not correctly overridden. (#35)

vChewing · May 25, 2022 · ce283cb · ce283cb
1 parent ea14d9d
commit ce283cb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/Sources/Megrez/4_Node.swift b/Sources/Megrez/4_Node.swift
@@ -84,6 +84,10 @@ extension Megrez {
         $0.score > $1.score
       }
 
+      if mutUnigrams.count > 0 {
+        mutScore = mutUnigrams[0].score
+      }
+
       for (i, gram) in mutUnigrams.enumerated() {
         mutValueUnigramIndexMap[gram.keyValue.value] = i
         mutCandidates.append(gram.keyValue)