pep8 fixes

Lightning-AI · Apr 17, 2020 · e0ddda5 · e0ddda5
1 parent 6908e03
commit e0ddda5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/metrics/functional/classification.py b/pytorch_lightning/metrics/functional/classification.py
@@ -387,4 +387,4 @@ def dice_score(pred: torch.Tensor, target: torch.Tensor, bg: bool = False,
             score_cls = (2 * tp).to(torch.float) / denom
 
         scores[i] += score_cls
-    return reduce(scores, reduction=reduction)
+    return reduce(scores, reduction=reduction)