Update Space (evaluate main: e4a27243)

pearsonr.py

@@ -13,6 +13,8 @@
 # limitations under the License.
 """Pearson correlation coefficient metric."""
 
+from dataclasses import dataclass
+
 import datasets
 from scipy.stats import pearsonr
 
@@ -83,13 +85,26 @@ doi = {10.1038/s41592-019-0686-2},
 """
 
 
+@dataclass
+class PearsonrConfig(evaluate.info.Config):
+
+    name: str = "default"
+
+    return_pvalue: bool = True
+
+
 @evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
 class Pearsonr(evaluate.Metric):
-    def _info(self):
+
+    CONFIG_CLASS = PearsonrConfig
+    ALLOWED_CONFIG_NAMES = ["default"]
+
+    def _info(self, config):
         return evaluate.MetricInfo(
             description=_DESCRIPTION,
             citation=_CITATION,
             inputs_description=_KWARGS_DESCRIPTION,
+            config=config,
             features=datasets.Features(
                 {
                     "predictions": datasets.Value("float"),
@@ -99,8 +114,8 @@ class Pearsonr(evaluate.Metric):
             reference_urls=["https://docs.scipy.org/doc/scipy/reference/generated/scipy.stats.pearsonr.html"],
         )
 
-    def _compute(self, predictions, references, return_pvalue=False):
-        if return_pvalue:
+    def _compute(self, predictions, references):
+        if self.config.return_pvalue:
             results = pearsonr(references, predictions)
             return {"pearsonr": results[0], "p-value": results[1]}
         else:

requirements.txt

@@ -1,2 +1,2 @@
-git+https://github.com/huggingface/evaluate@80448674f5447a9682afe051db243c4a13bfe4ff
+git+https://github.com/huggingface/evaluate@e4a2724377909fe2aeb4357e3971e5a569673b39
 scipy