diff --git a/src/aac_metrics/functional/bleu.py b/src/aac_metrics/functional/bleu.py index 82f9684..5a34039 100644 --- a/src/aac_metrics/functional/bleu.py +++ b/src/aac_metrics/functional/bleu.py @@ -219,7 +219,6 @@ def _bleu_compute( bleu_n_outs_sents[f"bleu_1_to_{n}"] = bleu_1_to_n_scores bleu_n_outs = bleu_n_outs_corpus, bleu_n_outs_sents - return bleu_n_outs else: return bleu_n_score diff --git a/src/aac_metrics/functional/meteor.py b/src/aac_metrics/functional/meteor.py index e395a6d..f1ed8d0 100644 --- a/src/aac_metrics/functional/meteor.py +++ b/src/aac_metrics/functional/meteor.py @@ -206,7 +206,7 @@ def meteor( "meteor": meteor_scores, } meteor_outs = meteor_outs_corpus, meteor_outs_sents - return meteor_outs + return meteor_outs # type: ignore else: return meteor_score diff --git a/src/aac_metrics/functional/rouge_l.py b/src/aac_metrics/functional/rouge_l.py index f23ae79..0af1d83 100644 --- a/src/aac_metrics/functional/rouge_l.py +++ b/src/aac_metrics/functional/rouge_l.py @@ -79,7 +79,8 @@ def _rouge_l_compute( rouge_l_outs_sents = { "rouge_l": rouge_l_scores_pt, } - return rouge_l_outs_corpus, rouge_l_outs_sents + rouge_l_outs = rouge_l_outs_corpus, rouge_l_outs_sents + return rouge_l_outs # type: ignore else: return rouge_l_score_pt diff --git a/src/aac_metrics/functional/sbert_sim.py b/src/aac_metrics/functional/sbert_sim.py index e73cbce..5adf67e 100644 --- a/src/aac_metrics/functional/sbert_sim.py +++ b/src/aac_metrics/functional/sbert_sim.py @@ -82,8 +82,8 @@ def sbert_sim( sbert_sim_outs_sents = { "sbert_sim": sbert_sim_scores, } - - return sbert_sim_outs_corpus, sbert_sim_outs_sents + sbert_sim_outs = sbert_sim_outs_corpus, sbert_sim_outs_sents + return sbert_sim_outs # type: ignore else: return sbert_sim_score diff --git a/src/aac_metrics/functional/spice.py b/src/aac_metrics/functional/spice.py index 5e1c083..6beeea9 100644 --- a/src/aac_metrics/functional/spice.py +++ b/src/aac_metrics/functional/spice.py @@ -219,7 +219,8 @@ def spice( spice_outs_sents = { "spice": spice_scores, } - return spice_outs_corpus, spice_outs_sents + spice_outs = spice_outs_corpus, spice_outs_sents + return spice_outs else: return spice_score diff --git a/src/aac_metrics/functional/vocab.py b/src/aac_metrics/functional/vocab.py index ff5b3cb..2782704 100644 --- a/src/aac_metrics/functional/vocab.py +++ b/src/aac_metrics/functional/vocab.py @@ -116,7 +116,8 @@ def vocab( "vocab.ratio_avg": vocab_len_ratio_avg, } - return corpus_scores, sents_scores + vocab_outs = corpus_scores, sents_scores + return vocab_outs # type: ignore def _corpus_vocab(tok_sents: list[list[str]], dtype: torch.dtype) -> Tensor: