From 0957bde92c64fd084b6fa4fc56613b91c9cda214 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Wed, 9 Aug 2023 17:23:06 +0100 Subject: [PATCH] Fix main --- test/test_transforms_v2_refactored.py | 14 +++----------- 1 file changed, 3 insertions(+), 11 deletions(-) diff --git a/test/test_transforms_v2_refactored.py b/test/test_transforms_v2_refactored.py index 9028b304c1b..c1a21b6346e 100644 --- a/test/test_transforms_v2_refactored.py +++ b/test/test_transforms_v2_refactored.py @@ -2240,17 +2240,9 @@ class TestPermuteChannels: def test_kernel(self, kernel, make_input, dtype, device): check_kernel(kernel, make_input(dtype=dtype, device=device), permutation=self._DEFAULT_PERMUTATION) - @pytest.mark.parametrize( - ("kernel", "make_input"), - [ - (F.permute_channels_image_tensor, make_image_tensor), - (F.permute_channels_image_pil, make_image_pil), - (F.permute_channels_image_tensor, make_image), - (F.permute_channels_video, make_video), - ], - ) - def test_dispatcher(self, kernel, make_input): - check_dispatcher(F.permute_channels, kernel, make_input(), permutation=self._DEFAULT_PERMUTATION) + @pytest.mark.parametrize("make_input", [make_image_tensor, make_image_pil, make_image, make_video]) + def test_dispatcher(self, make_input): + check_dispatcher(F.permute_channels, make_input(), permutation=self._DEFAULT_PERMUTATION) @pytest.mark.parametrize( ("kernel", "input_type"),