Resolve code style warning

Diuven · Jul 31, 2020 · 6df2018 · 6df2018
1 parent b157cfa
commit 6df2018
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/pytorch_lightning/metrics/functional/classification.py b/pytorch_lightning/metrics/functional/classification.py
@@ -205,9 +205,7 @@ def stat_scores_multiple_classes(
         fns = fns[:num_classes]
         sups = sups[:num_classes]
 
-        return tps, fps, tns, fns, sups
-
-    elif reduction == 'sum' or reduction == 'elementwise_mean':
+    if reduction == 'sum' or reduction == 'elementwise_mean':
         count_match_true = (pred == target).sum().float()
         oob_tp, oob_fp, oob_tn, oob_fn, oob_sup = stat_scores(pred, target, num_classes, argmax_dim)
 
@@ -224,11 +222,11 @@ def stat_scores_multiple_classes(
             tns /= num_classes
             sups /= num_classes
 
-        return tps, fps, tns, fns, sups
-
     else:
         raise ValueError("reduction type %s not supported" % reduction)
 
+    return tps, fps, tns, fns, sups
+
 
 def accuracy(
         pred: torch.Tensor,