Update Space (evaluate main: e4a27243)

mahalanobis.py

@@ -13,6 +13,9 @@
 # limitations under the License.
 """Mahalanobis metric."""
 
+from dataclasses import dataclass
+from typing import List, Optional
+
 import datasets
 import numpy as np
 
@@ -57,13 +60,25 @@ Examples:
 """
 
 
+@dataclass
+class MahalanobisConfig(evaluate.info.Config):
+
+    name: str = "default"
+
+    reference_distribution: Optional[List] = None
+
+
 @evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
 class Mahalanobis(evaluate.Metric):
-    def _info(self):
+    CONFIG_CLASS = MahalanobisConfig
+    ALLOWED_CONFIG_NAMES = ["default"]
+
+    def _info(self, config):
         return evaluate.MetricInfo(
             description=_DESCRIPTION,
             citation=_CITATION,
             inputs_description=_KWARGS_DESCRIPTION,
+            config=config,
             features=datasets.Features(
                 {
                     "X": datasets.Sequence(datasets.Value("float", id="sequence"), id="X"),
@@ -71,7 +86,12 @@ class Mahalanobis(evaluate.Metric):
             ),
         )
 
-    def _compute(self, X, reference_distribution):
+    def _compute(self, X):
+
+        if self.config.reference_distribution is None:
+            raise ValueError("You need to provide a `reference_distribution`.")
+        else:
+            reference_distribution = self.config.reference_distribution
 
         # convert to numpy arrays
         X = np.array(X)

requirements.txt

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