Vlasta commited on
Commit
b377cc7
1 Parent(s): 4fbdc4e

Fix input parameter type

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