fix tests

Lightning-AI · Mar 23, 2021 · c95f4e1 · c95f4e1
1 parent 5d5e852
commit c95f4e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchmetrics/functional/classification/auroc.py b/torchmetrics/functional/classification/auroc.py
@@ -75,7 +75,7 @@ def _auroc_compute(
  # calculate fpr, tpr
  if mode == 'multi-label':
  if average == AverageMethod.MICRO:
- fpr, tpr, _ = roc(preds.flatten(), target.flatten(), num_classes, pos_label, sample_weights)
+ fpr, tpr, _ = roc(preds.flatten(), target.flatten(), 1, pos_label, sample_weights)
  else:
  # for multilabel we iteratively evaluate roc in a binary fashion
  output = [