Skip to content

Commit f2b6f43

Browse files
authored
Fix main (#7816)
1 parent 1db6907 commit f2b6f43

File tree

1 file changed

+3
-11
lines changed

1 file changed

+3
-11
lines changed

test/test_transforms_v2_refactored.py

Lines changed: 3 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -2240,17 +2240,9 @@ class TestPermuteChannels:
22402240
def test_kernel(self, kernel, make_input, dtype, device):
22412241
check_kernel(kernel, make_input(dtype=dtype, device=device), permutation=self._DEFAULT_PERMUTATION)
22422242

2243-
@pytest.mark.parametrize(
2244-
("kernel", "make_input"),
2245-
[
2246-
(F.permute_channels_image_tensor, make_image_tensor),
2247-
(F.permute_channels_image_pil, make_image_pil),
2248-
(F.permute_channels_image_tensor, make_image),
2249-
(F.permute_channels_video, make_video),
2250-
],
2251-
)
2252-
def test_dispatcher(self, kernel, make_input):
2253-
check_dispatcher(F.permute_channels, kernel, make_input(), permutation=self._DEFAULT_PERMUTATION)
2243+
@pytest.mark.parametrize("make_input", [make_image_tensor, make_image_pil, make_image, make_video])
2244+
def test_dispatcher(self, make_input):
2245+
check_dispatcher(F.permute_channels, make_input(), permutation=self._DEFAULT_PERMUTATION)
22542246

22552247
@pytest.mark.parametrize(
22562248
("kernel", "input_type"),

0 commit comments

Comments
 (0)