Skip to content

Commit fa3ac89

Browse files
committed
Fix formatting
1 parent 5ab92db commit fa3ac89

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

segmentation_models_pytorch/metrics/functional.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -260,15 +260,15 @@ def _compute_metric(
260260
tn = tn.sum()
261261
score = metric_fn(tp, fp, fn, tn, **metric_kwargs)
262262

263-
elif reduction == "macro" :
263+
elif reduction == "macro":
264264
tp = tp.sum(0)
265265
fp = fp.sum(0)
266266
fn = fn.sum(0)
267267
tn = tn.sum(0)
268268
score = metric_fn(tp, fp, fn, tn, **metric_kwargs)
269269
score = _handle_zero_division(score, zero_division)
270270
score = (score * class_weights).mean()
271-
271+
272272
elif reduction == "weighted":
273273
tp = tp.sum(0)
274274
fp = fp.sum(0)

0 commit comments

Comments
 (0)