fix: passage map negatives when len(p)==1 (AnswerDotAI#21)

flexorRegev · Jan 7, 2024 · 16d0a89 · 16d0a89
1 parent bf3260c
commit 16d0a89
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ragatouille/data/training_data_processor.py b/ragatouille/data/training_data_processor.py
@@ -89,7 +89,7 @@ def _make_individual_triplets(self, query, positives, negatives):
  else:
  p = self.passage_map[positives[0]]
  for n in negatives:
- triplets.append([q, p, n])
+ triplets.append([q, p, self.passage_map[n]])
 
  return triplets