Made flag_reranker compute_score output type consistent.

muazhari · muazhari · commit 0216c33f5711 · 2024-04-09T09:53:08.000+07:00
diff --git a/FlagEmbedding/flag_reranker.py b/FlagEmbedding/flag_reranker.py
@@ -218,8 +218,6 @@ def compute_score(self, sentence_pairs: Union[List[Tuple[str, str]], Tuple[str,
         if normalize:
             all_scores = [sigmoid(score) for score in all_scores]
 
-        if len(all_scores) == 1:
-            return all_scores[0]
         return all_scores
 
 
