Fix formatting

Signed-off-by: Abdessamad <abdoomis6@gmail.com>

Author: AbdoMis

monai/transforms/utility/array.py

@@ -1070,7 +1070,11 @@ def __call__(self, img: NdarrayOrTensor) -> NdarrayOrTensor:
         if img.ndim == 4 and img.shape[0] == 1:
             img = img.squeeze(0)
 
-        result = [(img == 1) | (img == self.et_label), (img == 1) | (img == self.et_label) | (img == 2), img == self.et_label]
+        result = [
+            (img == 1) | (img == self.et_label),
+            (img == 1) | (img == self.et_label) | (img == 2),
+            img == self.et_label,
+        ]
         # merge labels 1 (tumor non-enh) and self.et_label (tumor enh) and 2 (large edema) to WT
         # self.et_label is ET (4 or 3)
         return torch.stack(result, dim=0) if isinstance(img, torch.Tensor) else np.stack(result, axis=0)

tests/transforms/test_convert_to_multi_channel.py

@@ -49,6 +49,7 @@
         ]
     )
 
+
 # Tests for et_label = 3
 for p in TEST_NDARRAYS:
     TESTS_ET_LABEL_3.extend(
@@ -66,6 +67,7 @@
         ]
     )
 
+
 class TestConvertToMultiChannel(unittest.TestCase):
     @parameterized.expand(TESTS)
     def test_type_shape(self, data, expected_result):
@@ -85,5 +87,6 @@ def test_invalid_et_label(self):
         with self.assertRaises(ValueError):
             ConvertToMultiChannelBasedOnBratsClasses(et_label=2)
 
+
 if __name__ == "__main__":
     unittest.main()