DarrenChensformer commited on
Commit
c9a7298
1 Parent(s): e3694cd

[fix] Change the expected type of input

Browse files
Files changed (1) hide show
  1. eval_keyphrase.py +2 -2
eval_keyphrase.py CHANGED
@@ -73,8 +73,8 @@ class eval_keyphrase(evaluate.Metric):
73
  inputs_description=_KWARGS_DESCRIPTION,
74
  # This defines the format of each prediction and reference
75
  features=datasets.Features({
76
- 'predictions': datasets.Value('int64'),
77
- 'references': datasets.Value('int64'),
78
  }),
79
  # Homepage of the module for documentation
80
  homepage="http://module.homepage",
 
73
  inputs_description=_KWARGS_DESCRIPTION,
74
  # This defines the format of each prediction and reference
75
  features=datasets.Features({
76
+ 'predictions': datasets.Sequence(datasets.Value('string')),
77
+ 'references': datasets.Sequence(datasets.Value('string')),
78
  }),
79
  # Homepage of the module for documentation
80
  homepage="http://module.homepage",