[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

monai/losses/unified_focal_loss.py

@@ -14,7 +14,6 @@
 import warnings
 
 import torch
-import torch.nn.functional as F
 from torch.nn.modules.loss import _Loss
 
 from monai.networks import one_hot
@@ -169,7 +168,7 @@ def forward(self, y_pred: torch.Tensor, y_true: torch.Tensor) -> torch.Tensor:
             back_ce = (1.0 - self.delta) * torch.pow(1.0 - y_pred[:, 0], self.gamma) * cross_entropy[:, 0]
             # (B, C-1, H, W)
             fore_ce = self.delta * cross_entropy[:, 1:]
-            
+
             loss = torch.cat([back_ce.unsqueeze(1), fore_ce], dim=1)  # (B, C, H, W)
 
         # Apply reduction
@@ -276,13 +275,13 @@ def forward(self, y_pred: torch.Tensor, y_true: torch.Tensor) -> torch.Tensor:
                 if y_true.shape[1] != 1:
                     y_true = y_true.unsqueeze(1)
                 y_true = one_hot(y_true, num_classes=n_pred_ch)
-        
+
         # Ensure y_true has the same shape as y_pred_act
         if y_true.shape != y_pred_act.shape:
              # This can happen if y_true is (B, H, W) and y_pred is (B, 1, H, W) after sigmoid
             if y_true.shape[1] != y_pred_act.shape[1] and y_true.ndim == y_pred_act.ndim - 1:
                 y_true = y_true.unsqueeze(1) # Add channel dim
-            
+
             if y_true.shape != y_pred_act.shape:
                 raise ValueError(f"ground truth has different shape ({y_true.shape}) from input ({y_pred_act.shape}) after activations/one-hot")
 
@@ -292,4 +291,4 @@ def forward(self, y_pred: torch.Tensor, y_true: torch.Tensor) -> torch.Tensor:
 
         loss: torch.Tensor = self.lambda_focal * f_loss + (1 - self.lambda_focal) * t_loss
 
-        return loss
+        return loss