chore: black

zhudotexe · Jun 19, 2024 · 7e741e4 · 7e741e4
1 parent 95deef8
commit 7e741e4
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/fanoutqa/eval/scorer.py b/fanoutqa/eval/scorer.py
@@ -134,9 +134,12 @@ def score_rouge(self) -> Tuple[RougeScore, Dict[str, RougeScore]]:
             results = self.rouge.score(str_answer(q.answer), str_answer(a["answer"]))
             for k, v in results.items():
                 scores[k].append(v)
-            raw_scores[q.id] = RougeScore(**{
-                k: RougeScorePart(precision=v.precision, recall=v.recall, fscore=v.fmeasure) for k, v in results.items()
-            })
+            raw_scores[q.id] = RougeScore(
+                **{
+                    k: RougeScorePart(precision=v.precision, recall=v.recall, fscore=v.fmeasure)
+                    for k, v in results.items()
+                }
+            )
 
         assert all(len(v) == self.eval_len for v in scores.values())
         assert len(raw_scores) == self.eval_len