Skip to content

Commit

Permalink
fix linter
Browse files Browse the repository at this point in the history
  • Loading branch information
kprokofi committed Dec 18, 2024
1 parent e51a403 commit b44ebea
Show file tree
Hide file tree
Showing 2 changed files with 20 additions and 4 deletions.
12 changes: 10 additions & 2 deletions tests/unit/core/model/test_detection_3d.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)),
],
Expand Down
12 changes: 10 additions & 2 deletions tests/unit/core/model/test_keypoint_detection.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)),
],
Expand Down

0 comments on commit b44ebea

Please sign in to comment.