Fix flake8 error

Signed-off-by: Suraj Pai <[email protected]>
surajpaib · Sep 6, 2024 · 27a0c58 · 27a0c58
1 parent bf0c5f0
commit 27a0c58
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/test_compute_generalized_dice.py b/tests/test_compute_generalized_dice.py
@@ -133,7 +133,8 @@
 ]
 
 
-TEST_CASE_10 = [  # y (2, 3, 2, 2) y_pred (2, 3, 2, 2) expected out (2, 3) with compute_generalized_dice and (3) with GeneralizedDiceScore "mean_batch"
+TEST_CASE_10 = [  # y (2, 3, 2, 2) y_pred (2, 3, 2, 2) expected out (2, 3) with compute_generalized_dice
+    # and (3) with GeneralizedDiceScore "mean_batch"
     {
         "y_pred": torch.tensor(
             [
@@ -152,7 +153,8 @@
     [[0.857143, 0.0, 0.0], [0.5, 0.4, 0.666667]],
 ]
 
-TEST_CASE_11 = [  # y (2, 3, 2, 2) y_pred (2, 3, 2, 2) expected out (2, 1) with compute_generalized_dice (summed over classes) and (2) with GeneralizedDiceScore "mean_channel"
+TEST_CASE_11 = [  # y (2, 3, 2, 2) y_pred (2, 3, 2, 2) expected out (2, 1) with compute_generalized_dice (summed over classes)
+    # and (2) with GeneralizedDiceScore "mean_channel"
     {
         "y_pred": torch.tensor(
             [