brian920128 commited on
Commit
6815104
1 Parent(s): edc5524
Files changed (1) hide show
  1. doc_retrieve_metrics.py +2 -2
doc_retrieve_metrics.py CHANGED
@@ -71,8 +71,8 @@ class DocRetrieveMetrics(evaluate.Metric):
71
  inputs_description=_KWARGS_DESCRIPTION,
72
  # This defines the format of each prediction and reference
73
  features=datasets.Features({
74
- 'predictions': datasets.Value('int64'),
75
- 'references': datasets.Value('int64'),
76
  }),
77
  # Homepage of the module for documentation
78
  homepage="http://module.homepage",
 
71
  inputs_description=_KWARGS_DESCRIPTION,
72
  # This defines the format of each prediction and reference
73
  features=datasets.Features({
74
+ "predictions": datasets.Sequence(datasets.Sequence(datasets.Value("string"))),
75
+ "references": datasets.Sequence(datasets.Sequence(datasets.Value("string"))),
76
  }),
77
  # Homepage of the module for documentation
78
  homepage="http://module.homepage",