Skip to content

Commit

Permalink
fix ut
Browse files Browse the repository at this point in the history
  • Loading branch information
Tau-J committed Oct 9, 2023
1 parent c130389 commit 9577e15
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions tests/test_evaluation/test_functional/test_transforms.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,12 +49,8 @@ def test_transform_pred(self):
3,
)),
keypoint_scores=np.ones((1, 17)))
kpt_info_copy = deepcopy(kpt_info)

_ = transform_pred(kpt_info, num_keypoints, mapping)

self.assertEqual(kpt_info['num_keypoints'], 5)
self.assertEqual(len(kpt_info['keypoints']), 1)
for i, j in mapping:
self.assertListEqual(kpt_info_copy['keypoints'][:, i, :],
kpt_info['keypoints'][:, j, :])

0 comments on commit 9577e15

Please sign in to comment.