@@ -2240,17 +2240,9 @@ class TestPermuteChannels:
2240
2240
def test_kernel (self , kernel , make_input , dtype , device ):
2241
2241
check_kernel (kernel , make_input (dtype = dtype , device = device ), permutation = self ._DEFAULT_PERMUTATION )
2242
2242
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 )
2254
2246
2255
2247
@pytest .mark .parametrize (
2256
2248
("kernel" , "input_type" ),
0 commit comments