swrwa

app.py

@@ -5,6 +5,6 @@ from evaluate.utils import launch_gradio_widget
 module = evaluate.load("brian920128/doc_retrieve_metrics")
 
 
-result = module.compute(references=[["e", ""], ["", ""]], predictions=[["e", ""], ["", ""]])
+result = module.compute(references=[[["e", ""], ["", ""]]], predictions=[[["e", ""], ["", ""]]])
 print(result)
 #launch_gradio_widget(module)

doc_retrieve_metrics.py

@@ -118,8 +118,8 @@ class DocRetrieveMetrics(evaluate.Metric):
             inputs_description=_KWARGS_DESCRIPTION,
             # This defines the format of each prediction and reference
             features=datasets.Features({
-                "predictions": datasets.Sequence(datasets.Sequence(datasets.Value("string"))),
-                "references": datasets.Sequence(datasets.Sequence(datasets.Value("string"))),
+                "predictions": datasets.Sequence(datasets.Value("string")),
+                "references": datasets.Sequence(datasets.Value("string")),
             }),
             # Homepage of the module for documentation
             homepage="http://module.homepage",