Natooz commited on
Commit
4fb9244
·
unverified ·
1 Parent(s): 9309342

fix input format

Browse files
Files changed (1) hide show
  1. ece.py +1 -1
ece.py CHANGED
@@ -76,7 +76,7 @@ class ECE(evaluate.Metric):
76
  inputs_description=_KWARGS_DESCRIPTION,
77
  # This defines the format of each prediction and reference
78
  features=datasets.Features({
79
- 'predictions': datasets.Value('float32'),
80
  'references': datasets.Value('int64'),
81
  }),
82
  # Homepage of the module for documentation
 
76
  inputs_description=_KWARGS_DESCRIPTION,
77
  # This defines the format of each prediction and reference
78
  features=datasets.Features({
79
+ 'predictions': datasets.Sequence(datasets.Value('float32')),
80
  'references': datasets.Value('int64'),
81
  }),
82
  # Homepage of the module for documentation