Skip to content

Commit 87d05c6

Browse files
authored
Remove warning filter in transforms test (#8523)
1 parent 8f9d810 commit 87d05c6

File tree

1 file changed

+1
-9
lines changed

1 file changed

+1
-9
lines changed

test/test_transforms_v2.py

+1-9
Original file line numberDiff line numberDiff line change
@@ -394,15 +394,7 @@ def _check_transform_sample_input_smoke(transform, input, *, adapter):
394394

395395

396396
def check_transform(transform, input, check_v1_compatibility=True, check_sample_input=True):
397-
# TODO: remove this cm once https://github.com/pytorch/vision/issues/8517
398-
# and https://github.com/pytorch/pytorch/issues/130242 are resolved.
399-
if isinstance(transform, (transforms.RandomResizedCrop, transforms.LinearTransformation)):
400-
cm = pytest.warns(FutureWarning, match="You are using `torch.load`")
401-
else:
402-
cm = contextlib.nullcontext()
403-
404-
with cm:
405-
pickle.loads(pickle.dumps(transform))
397+
pickle.loads(pickle.dumps(transform))
406398

407399
output = transform(input)
408400
assert isinstance(output, type(input))

0 commit comments

Comments
 (0)