Fix sentence align bug

ProbablyFaiz · Jun 15, 2024 · f53ce00 · f53ce00
1 parent 2b3870e
commit f53ce00
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rl/utils/locate_passage.py b/rl/utils/locate_passage.py
@@ -116,8 +116,8 @@ def get_best_ngram_match(
             )
             for sent in para["sentences"]
         ]
-        start = min(sentences, key=lambda s: abs(s.start - start))["start"]
-        end = max(sentences, key=lambda s: abs(s.end - end))["end"]
+        start = min(sentences, key=lambda s: abs(s["start"] - start))["start"]
+        end = max(sentences, key=lambda s: abs(s["end"] - end))["end"]
 
     return start, end