Skip to content

Commit ce817c1

Browse files
committed
Fix formatting in array.py
Signed-off-by: Abdessamad <abdoomis6@gmail.com>
1 parent 48c00c0 commit ce817c1

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

monai/transforms/utility/array.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1070,7 +1070,11 @@ def __call__(self, img: NdarrayOrTensor) -> NdarrayOrTensor:
10701070
if img.ndim == 4 and img.shape[0] == 1:
10711071
img = img.squeeze(0)
10721072

1073-
result = [(img == 1) | (img == self.et_label), (img == 1) | (img == self.et_label) | (img == 2), img == self.et_label]
1073+
result = [
1074+
(img == 1) | (img == self.et_label),
1075+
(img == 1) | (img == self.et_label) | (img == 2),
1076+
img == self.et_label,
1077+
]
10741078
# merge labels 1 (tumor non-enh) and self.et_label (tumor enh) and 2 (large edema) to WT
10751079
# self.et_label is ET (4 or 3)
10761080
return torch.stack(result, dim=0) if isinstance(img, torch.Tensor) else np.stack(result, axis=0)

0 commit comments

Comments
 (0)