output type

Author: staoxiao

FlagEmbedding/flag_reranker.py

@@ -367,8 +367,8 @@ 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]
+        # if len(all_scores) == 1:
+        #     return all_scores[0]
 
         return all_scores
 
@@ -560,10 +560,10 @@ def compute_score(self, sentence_pairs: Union[List[Tuple[str, str]], Tuple[str,
             if normalize:
                 all_scores[i] = [sigmoid(score) for score in all_scores[i]]
 
-        if len(all_scores) == 1:
-            if len(all_scores[0]) == 1:
-                return all_scores[0][0]
-            return all_scores[0]
+        # if len(all_scores) == 1:
+        #     if len(all_scores[0]) == 1:
+        #         return all_scores[0][0]
+        #     return all_scores[0]
 
         return all_scores
 

LM_Cocktail/setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name='LM_Cocktail',
-    version='0.0.4',
+    version='0.0.5',
     description='LM_Cocktail',
     long_description=readme,
     long_description_content_type="text/markdown",