Skip to content

Commit 30ad075

Browse files
committed
Update test_prototype_transforms_functional.py
1 parent 2ece9af commit 30ad075

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

test/test_prototype_transforms_functional.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -345,7 +345,7 @@ def crop_segmentation_mask():
345345
width=width,
346346
)
347347

348-
348+
349349
@register_kernel_info_from_sample_inputs_fn
350350
def vertical_flip_segmentation_mask():
351351
for mask in make_segmentation_masks():
@@ -933,4 +933,3 @@ def test_correctness_vertical_flip_segmentation_mask_on_fixed_input(device):
933933
expected_mask = torch.zeros((3, 3, 3), dtype=torch.long, device=device)
934934
expected_mask[:, -1, :] = 1
935935
torch.testing.assert_close(out_mask, expected_mask)
936-

0 commit comments

Comments
 (0)