Skip to content

GitLab

Unverified Commit f2b6f43a authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Fix main (#7816)

parent 1db6907b
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 11 deletions
test/test_transforms_v2_refactored.py
View file @ f2b6f43a
...@@ -2240,17 +2240,9 @@ class TestPermuteChannels: ...@@ -2240,17 +2240,9 @@ class TestPermuteChannels:
def test_kernel(self, kernel, make_input, dtype, device): def test_kernel(self, kernel, make_input, dtype, device):
check_kernel(kernel, make_input(dtype=dtype, device=device), permutation=self._DEFAULT_PERMUTATION) check_kernel(kernel, make_input(dtype=dtype, device=device), permutation=self._DEFAULT_PERMUTATION)
@pytest.mark.parametrize( @pytest.mark.parametrize("make_input", [make_image_tensor, make_image_pil, make_image, make_video])
("kernel", "make_input"), def test_dispatcher(self, make_input):
[ check_dispatcher(F.permute_channels, make_input(), permutation=self._DEFAULT_PERMUTATION)
(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( @pytest.mark.parametrize(
("kernel", "input_type"), ("kernel", "input_type"),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment