jordyvl commited on
Commit
3e6ca8f
1 Parent(s): adde0a6

update app

Browse files
Files changed (1) hide show
  1. ece.py +1 -1
ece.py CHANGED
@@ -218,7 +218,7 @@ class ECE(evaluate.EvaluationModule):
218
  inputs_description=_KWARGS_DESCRIPTION,
219
  features=datasets.Features(
220
  {
221
- "predictions": datasets.Array2D("float32"), # or Sequence
222
  "references": datasets.Sequence(datasets.Value("int64")),
223
  }
224
  ),
 
218
  inputs_description=_KWARGS_DESCRIPTION,
219
  features=datasets.Features(
220
  {
221
+ "predictions": datasets.Sequence(datasets.Value("float32")),
222
  "references": datasets.Sequence(datasets.Value("int64")),
223
  }
224
  ),