fix type

top5_error_rate.py

@@ -44,12 +44,12 @@ class Top5ErrorRate(evaluate.Metric):
             inputs_description=_KWARGS_DESCRIPTION,
             features=datasets.Features(
                 {
-                    "predictions": datasets.Sequence(datasets.Value("float32")),
+                    "predictions": datasets.Sequence(datasets.Sequence(datasets.Value("float32"))),
                     "references": datasets.Sequence(datasets.Value("int32")),
                 }
                 if self.config_name == "multilabel"
                 else {
-                    "predictions": datasets.Value("float32"),
+                    "predictions": datasets.Sequence(datasets.Value("float32")),
                     "references": datasets.Value("int32"),
                 }
             ),
@@ -63,13 +63,9 @@ class Top5ErrorRate(evaluate.Metric):
             references: list[int] = None,
             **kwargs,
     ) -> Dict[str, Any]:
-        print(predictions)
-        print(references)
         # to numpy array
         outputs = np.array(predictions, dtype=np.float32)
         labels = np.array(references)
-        print(outputs)
-        print(labels)
 
         # Top-1 ACC
         pred = outputs.argmax(axis=1)