Skip to content

Commit 3441ab9

Browse files
committed
Add tests for perspective
1 parent a43e534 commit 3441ab9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/test_transforms_v2.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)