diff --git a/test/smoke_test.py b/test/smoke_test.py index 9ffc9117773..35e079c31d1 100644 --- a/test/smoke_test.py +++ b/test/smoke_test.py @@ -27,6 +27,7 @@ def smoke_test_torchvision_read_decode() -> None: if img_png.ndim != 3 or img_png.numel() < 100: raise RuntimeError(f"Unexpected shape of img_png: {img_png.shape}") + def smoke_test_compile() -> None: model = resnet50().cuda() model = torch.compile(model) @@ -34,6 +35,7 @@ def smoke_test_compile() -> None: out = model(x) print(f"torch.compile model output: {out.shape}") + def smoke_test_torchvision_resnet50_classify(device: str = "cpu") -> None: img = read_image(str(SCRIPT_DIR / ".." / "gallery" / "assets" / "dog2.jpg")).to(device) @@ -73,6 +75,5 @@ def main() -> None: smoke_test_torchvision_resnet50_classify("mps") - if __name__ == "__main__": main() diff --git a/test/test_transforms.py b/test/test_transforms.py index 03b385e9edd..67efee84515 100644 --- a/test/test_transforms.py +++ b/test/test_transforms.py @@ -2,6 +2,7 @@ import os import random import re +import sys import textwrap import warnings from functools import partial @@ -2268,6 +2269,7 @@ def test_random_grayscale_with_grayscale_input(): # TODO: remove in 0.17 when we can delete functional_pil.py and functional_tensor.py +@pytest.mark.xfail(sys.version_info[:2] == (3, 10), reason="See #7372") @pytest.mark.parametrize( "import_statement", ( diff --git a/test/test_transforms_v2.py b/test/test_transforms_v2.py index f5ca976963a..b83f3542349 100644 --- a/test/test_transforms_v2.py +++ b/test/test_transforms_v2.py @@ -2,6 +2,7 @@ import pathlib import random import re +import sys import textwrap import warnings from collections import defaultdict @@ -2087,6 +2088,7 @@ def test_sanitize_bounding_boxes_errors(): transforms.SanitizeBoundingBox()(different_sizes) +@pytest.mark.xfail(sys.version_info[:2] == (3, 10), reason="See #7372") @pytest.mark.parametrize( "import_statement", ( @@ -2120,6 +2122,7 @@ def test_warnings_v2_namespaces(import_statement, call_disable_warning): assert_run_python_script(textwrap.dedent(source)) +@pytest.mark.xfail(sys.version_info[:2] == (3, 10), reason="See #7372") def test_no_warnings_v1_namespace(): source = """ import warnings