Skip to content

Commit 074974b

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent fa26b09 commit 074974b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

monai/transforms/post/array.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -681,15 +681,15 @@ def __init__(self, weights: Sequence[float] | NdarrayOrTensor | None = None) ->
681681
def __call__(self, img: Sequence[NdarrayOrTensor] | NdarrayOrTensor) -> NdarrayOrTensor:
682682
out_pt = None
683683
total_weight = 0.0
684-
684+
685685
for i, pred in enumerate(img):
686686
pred = torch.as_tensor(pred)
687687
if out_pt is None:
688688
out_pt = torch.zeros_like(pred)
689689

690690
if self.weights is not None:
691691
weight = self.weights[i].to(pred.device)
692-
692+
693693
out_pt += pred * weight
694694
total_weight += weight
695695

0 commit comments

Comments
 (0)