fixed duplicate calculation of spearmanr function in metrics wrapper. (…

…#176)
huggingface · Jul 7, 2022 · 4f4e17d · 4f4e17d
1 parent f62ad7c
commit 4f4e17d
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/metrics/spearmanr/spearmanr.py b/metrics/spearmanr/spearmanr.py
@@ -113,12 +113,8 @@ def _info(self):
         )
 
     def _compute(self, predictions, references, return_pvalue=False):
+        results = spearmanr(references, predictions)
         if return_pvalue:
-            return {
-                "spearmanr": spearmanr(references, predictions)[0],
-                "spearmanr_pvalue": spearmanr(references, predictions)[1],
-            }
+            return {"spearmanr": results[0], "spearmanr_pvalue": results[1]}
         else:
-            return {
-                "spearmanr": spearmanr(references, predictions)[0],
-            }
+            return {"spearmanr": results[0]}