add lint fixes

Signed-off-by: Soumya Snigdha Kundu <soumya_snigdha.kundu@kcl.ac.uk>

Author: aymuos15

monai/metrics/meandice.py

@@ -311,7 +311,7 @@ def __call__(self, y_pred: torch.Tensor, y: torch.Tensor) -> torch.Tensor | tupl
         if y_pred.shape[1] == 1 and n_pred_ch > 1:
             y_pred_bool = torch.zeros(batch_size, n_pred_ch, *y_pred.shape[2:], dtype=torch.bool, device=device)
             for c in range(n_pred_ch):
-                y_pred_bool[:, c] = (y_pred[:, 0] == c)
+                y_pred_bool[:, c] = y_pred[:, 0] == c
         else:
             y_pred_bool = y_pred.bool()
 
@@ -345,9 +345,12 @@ def __call__(self, y_pred: torch.Tensor, y: torch.Tensor) -> torch.Tensor | tupl
             empty_mask = y_sum == 0
             dice = torch.where(
                 empty_mask,
-                torch.where(pred_sum == 0, torch.tensor(1.0, device=device, dtype=dice.dtype),
-                           torch.tensor(0.0, device=device, dtype=dice.dtype)),
-                dice
+                torch.where(
+                    pred_sum == 0,
+                    torch.tensor(1.0, device=device, dtype=dice.dtype),
+                    torch.tensor(0.0, device=device, dtype=dice.dtype),
+                ),
+                dice,
             )
 
         # Select channels: exclude background if requested

tests/metrics/test_compute_meandice.py

@@ -291,12 +291,7 @@
                 [[[0.0, 0.0], [0.0, 1.0]], [[1.0, 0.0], [0.0, 0.0]], [[0.0, 1.0], [1.0, 0.0]]],
             ]
         ),  # (2, 3, 2, 2) one-hot
-        "y": torch.tensor(
-            [
-                [[[0.0, 0.0], [0.0, 1.0]]],
-                [[[0.0, 0.0], [0.0, 1.0]]],
-            ]
-        ),  # (2, 1, 2, 2) class indices
+        "y": torch.tensor([[[[0.0, 0.0], [0.0, 1.0]]], [[[0.0, 0.0], [0.0, 1.0]]]]),  # (2, 1, 2, 2) class indices
         "include_background": False,
     },
     # batch 0: class 1 y_gt=[[0,0],[0,1]], y_pred=[[0,0],[1,1]] -> dice=2/3