@@ -4854,7 +4854,7 @@ def test_kernel_video(self):
48544854
48554855 @pytest .mark .parametrize (
48564856 "make_input" ,
4857- [make_image_tensor , make_image_pil , make_image , make_bounding_boxes , make_segmentation_mask , make_video ],
4857+ [make_image_tensor , make_image_pil , make_image , make_bounding_boxes , make_segmentation_mask , make_video , make_keypoints ],
48584858 )
48594859 def test_functional (self , make_input ):
48604860 check_functional (F .perspective , make_input (), ** self .MINIMAL_KWARGS )
@@ -4877,7 +4877,7 @@ def test_functional_signature(self, kernel, input_type):
48774877 @pytest .mark .parametrize ("distortion_scale" , [0.5 , 0.0 , 1.0 ])
48784878 @pytest .mark .parametrize (
48794879 "make_input" ,
4880- [make_image_tensor , make_image_pil , make_image , make_bounding_boxes , make_segmentation_mask , make_video ],
4880+ [make_image_tensor , make_image_pil , make_image , make_bounding_boxes , make_segmentation_mask , make_video , make_keypoints ],
48814881 )
48824882 def test_transform (self , distortion_scale , make_input ):
48834883 check_transform (transforms .RandomPerspective (distortion_scale = distortion_scale , p = 1 ), make_input ())
0 commit comments