Skip to content

Commit

Permalink
fix-issue-6366
Browse files Browse the repository at this point in the history
Signed-off-by: staydelight <kevin295643815697236@gmail.com>
  • Loading branch information
staydelight committed Oct 11, 2024
1 parent 074974b commit bae8f59
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion monai/transforms/post/array.py
Original file line number Diff line number Diff line change
Expand Up @@ -694,7 +694,8 @@ def __call__(self, img: Sequence[NdarrayOrTensor] | NdarrayOrTensor) -> NdarrayO
total_weight += weight

out_pt /= total_weight
return post_convert(out_pt, img)
return self.post_convert(out_pt, img)


class VoteEnsemble(Ensemble, Transform):
"""
Expand Down

0 comments on commit bae8f59

Please sign in to comment.