Skip to content

Commit a9b8daf

Browse files
committed
Avoids skipping some tests
1 parent b7b1576 commit a9b8daf

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

test/test_transforms_v2.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -663,8 +663,7 @@ def test_kernel_video(self):
663663
[make_image_tensor, make_image_pil, make_image, make_bounding_boxes, make_segmentation_mask, make_video],
664664
)
665665
def test_functional(self, size, make_input):
666-
if not (max_size_kwarg := self._make_max_size_kwarg(use_max_size=True, size=size)):
667-
return
666+
max_size_kwarg = self._make_max_size_kwarg(use_max_size=size is None, size=size)
668667

669668
check_functional(
670669
F.resize,
@@ -704,8 +703,7 @@ def test_functional_signature(self, kernel, input_type):
704703
],
705704
)
706705
def test_transform(self, size, device, make_input):
707-
if not (max_size_kwarg := self._make_max_size_kwarg(use_max_size=True, size=size)):
708-
return
706+
max_size_kwarg = self._make_max_size_kwarg(use_max_size=size is None, size=size)
709707

710708
check_transform(
711709
transforms.Resize(size=size, **max_size_kwarg, antialias=True),

0 commit comments

Comments
 (0)