diff --git a/tests/unit/core/types/test_label.py b/tests/unit/core/types/test_label.py index 8b853e4c4c..6789423494 100644 --- a/tests/unit/core/types/test_label.py +++ b/tests/unit/core/types/test_label.py @@ -20,7 +20,11 @@ def test_seg_label_info(): ["0", "1", "2"], [["label_0", "label_1", "label_2"]], ) - assert SegLabelInfo.from_num_classes(1) == SegLabelInfo(["background", "label_0"], ["0", "1"], [["background", "label_0"]]) + assert SegLabelInfo.from_num_classes(1) == SegLabelInfo( + ["background", "label_0"], + ["0", "1"], + [["background", "label_0"]], + ) assert SegLabelInfo.from_num_classes(0) == NullLabelInfo()