diff --git a/tests/unit/core/model/test_detection_3d.py b/tests/unit/core/model/test_detection_3d.py index 673133a456..c070305c09 100644 --- a/tests/unit/core/model/test_detection_3d.py +++ b/tests/unit/core/model/test_detection_3d.py @@ -62,8 +62,16 @@ def test_export_parameters(self, model): ("label_info", "expected_label_info"), [ ( - LabelInfo(label_names=["label1", "label2", "label3"], label_groups=[["label1", "label2", "label3"]], label_ids=["0", "1", "2"]), - LabelInfo(label_names=["label1", "label2", "label3"], label_groups=[["label1", "label2", "label3"]], label_ids=["0", "1", "2"]), + LabelInfo( + label_names=["label1", "label2", "label3"], + label_groups=[["label1", "label2", "label3"]], + label_ids=["0", "1", "2"], + ), + LabelInfo( + label_names=["label1", "label2", "label3"], + label_groups=[["label1", "label2", "label3"]], + label_ids=["0", "1", "2"], + ), ), (LabelInfo.from_num_classes(num_classes=5), LabelInfo.from_num_classes(num_classes=5)), ], diff --git a/tests/unit/core/model/test_keypoint_detection.py b/tests/unit/core/model/test_keypoint_detection.py index 77e19d605e..ca32686e76 100644 --- a/tests/unit/core/model/test_keypoint_detection.py +++ b/tests/unit/core/model/test_keypoint_detection.py @@ -62,8 +62,16 @@ def test_export_parameters(self, model): ("label_info", "expected_label_info"), [ ( - LabelInfo(label_names=["label1", "label2", "label3"], label_groups=[["label1", "label2", "label3"]], label_ids=["0", "1", "2"]), - LabelInfo(label_names=["label1", "label2", "label3"], label_groups=[["label1", "label2", "label3"]], label_ids=["0", "1", "2"]), + LabelInfo( + label_names=["label1", "label2", "label3"], + label_groups=[["label1", "label2", "label3"]], + label_ids=["0", "1", "2"], + ), + LabelInfo( + label_names=["label1", "label2", "label3"], + label_groups=[["label1", "label2", "label3"]], + label_ids=["0", "1", "2"], + ), ), (LabelInfo.from_num_classes(num_classes=5), LabelInfo.from_num_classes(num_classes=5)), ],