Properly formatted files

Signed-off-by: José Morano <[email protected]>

Author: José Morano

monai/losses/focal_loss.py

@@ -274,7 +274,7 @@ def focal_loss_with_probs(
 
     # Compute focal loss components
     log_pt = torch.log(torch.clamp(pt, min=1e-8))  # Avoid log(0)
-    focal_factor = (1 - pt).pow(gamma)             # (1 - pt)**gamma
+    focal_factor = (1 - pt).pow(gamma)  # (1 - pt)**gamma
 
     loss = -focal_factor * log_pt
 

tests/losses/test_focal_loss.py

@@ -311,7 +311,9 @@ def test_bin_seg_3d(self):
                 # The prediction here are probabilities, not logits.
                 pred_very_good = target_one_hot.clone().float()
             else:
-                pred_very_good = 1000 * F.one_hot(target, num_classes=num_classes).permute(0, 4, 1, 2, 3).float() - 500.0
+                pred_very_good = (
+                    1000 * F.one_hot(target, num_classes=num_classes).permute(0, 4, 1, 2, 3).float() - 500.0
+                )
 
             # initialize the mean dice loss
             loss = FocalLoss(to_onehot_y=True, use_softmax=use_softmax, use_sigmoid=use_sigmoid)