diff --git a/tests/unittests/segmentation/test_dice.py b/tests/unittests/segmentation/test_dice.py index b1f3a3e6c8d..a3df3105b3e 100644 --- a/tests/unittests/segmentation/test_dice.py +++ b/tests/unittests/segmentation/test_dice.py @@ -147,9 +147,9 @@ def test_zero_division(zero_division, average): }, } - assert torch.allclose( - score, res_dict[average][zero_division], atol=1e-4, equal_nan=True - ), f"Expected {res_dict[average][zero_division]} but got {score}" + assert torch.allclose(score, res_dict[average][zero_division], atol=1e-4, equal_nan=True), ( + f"Expected {res_dict[average][zero_division]} but got {score}" + ) @pytest.mark.parametrize("compute_groups", [True, False])