fix ut

open-mmlab · Oct 9, 2023 · c130389 · c130389
1 parent d5ca650
commit c130389
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_evaluation/test_functional/test_transforms.py b/tests/test_evaluation/test_functional/test_transforms.py
@@ -47,7 +47,8 @@ def test_transform_pred(self):
                 1,
                 17,
                 3,
-            )))
+            )),
+            keypoint_scores=np.ones((1, 17)))
         kpt_info_copy = deepcopy(kpt_info)
 
         _ = transform_pred(kpt_info, num_keypoints, mapping)