diff --git a/src/aac_metrics/functional/bert_score_mrefs.py b/src/aac_metrics/functional/bert_score_mrefs.py index a70fd47..d5ab544 100644 --- a/src/aac_metrics/functional/bert_score_mrefs.py +++ b/src/aac_metrics/functional/bert_score_mrefs.py @@ -109,7 +109,9 @@ def bert_score_mrefs( tfmers_logging.set_verbosity(tfmers_verbosity) # note: torchmetrics returns a float if input contains 1 cand and 1 ref, even in list - if len(duplicated_cands) == 1 and all(isinstance(v, float) for v in sents_scores.values()): + if len(duplicated_cands) == 1 and all( + isinstance(v, float) for v in sents_scores.values() + ): sents_scores = {k: [v] for k, v in sents_scores.items()} # sents_scores keys: "precision", "recall", "f1"