update

ranking_loss.py

@@ -15,7 +15,6 @@
 
 import evaluate
 import datasets
-from typing import Any
 from sklearn.metrics import label_ranking_loss
 
 # TODO: Add BibTeX citation
@@ -72,8 +71,8 @@ class RankingLoss(evaluate.Metric):
             inputs_description=_KWARGS_DESCRIPTION,
             # This defines the format of each prediction and reference
             features=datasets.Features({
-                'predictions': Any,
-                'references': Any,
+                'predictions': datasets.Sequence(datasets.Value('float32')),
+                'references': datasets.Sequence(datasets.Value('int64')),
             }),
             # Homepage of the module for documentation
             # homepage="http://module.homepage",