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:
image = make_image_tensor(dtype=dtype, device=device)
check_kernel(
F.elastic_image_tensor,
F.elastic_image,
image,
displacement=self._make_displacement(image),
**{param: value},
......@@ -2320,9 +2320,9 @@ class TestElastic:
@pytest.mark.parametrize(
("kernel", "input_type"),
[
(F.elastic_image_tensor, torch.Tensor),
(F.elastic_image_pil, PIL.Image.Image),
(F.elastic_image_tensor, datapoints.Image),
(F.elastic_image, torch.Tensor),
(F._elastic_image_pil, PIL.Image.Image),
(F.elastic_image, datapoints.Image),
(F.elastic_bounding_boxes, datapoints.BoundingBoxes),
(F.elastic_mask, datapoints.Mask),
(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