Mod: Update internal functional typing.

Labbeti · May 27, 2024 · 62acc11 · 62acc11
1 parent d78c8b9
commit 62acc11
Show file tree

Hide file tree

Showing 6 changed files with 9 additions and 7 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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: