Skip to content

GitLab

Unverified Commit 4cba51c5 authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

fix elastic tests (#7841)

parent bda807d5
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
test/test_transforms_v2_refactored.py
View file @ 4cba51c5
...@@ -2279,7 +2279,7 @@ class TestElastic: ...@@ -2279,7 +2279,7 @@ class TestElastic:
image = make_image_tensor(dtype=dtype, device=device) image = make_image_tensor(dtype=dtype, device=device)
check_kernel( check_kernel(
F.elastic_image_tensor, F.elastic_image,
image, image,
displacement=self._make_displacement(image), displacement=self._make_displacement(image),
**{param: value}, **{param: value},
...@@ -2320,9 +2320,9 @@ class TestElastic: ...@@ -2320,9 +2320,9 @@ class TestElastic:
@pytest.mark.parametrize( @pytest.mark.parametrize(
("kernel", "input_type"), ("kernel", "input_type"),
[ [
(F.elastic_image_tensor, torch.Tensor), (F.elastic_image, torch.Tensor),
(F.elastic_image_pil, PIL.Image.Image), (F._elastic_image_pil, PIL.Image.Image),
(F.elastic_image_tensor, datapoints.Image), (F.elastic_image, datapoints.Image),
(F.elastic_bounding_boxes, datapoints.BoundingBoxes), (F.elastic_bounding_boxes, datapoints.BoundingBoxes),
(F.elastic_mask, datapoints.Mask), (F.elastic_mask, datapoints.Mask),
(F.elastic_video, datapoints.Video), (F.elastic_video, datapoints.Video),
......
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