diff --git a/python/tvm/testing/utils.py b/python/tvm/testing/utils.py index 96275e2af66f..054257e07aa2 100644 --- a/python/tvm/testing/utils.py +++ b/python/tvm/testing/utils.py @@ -97,7 +97,8 @@ def test_something(): IS_IN_CI = os.getenv("CI", "") == "true" skip_if_wheel_test = pytest.mark.skipif( - os.getenv("WHEEL_TEST") is not None, reason="Test not supported in wheel." + os.getenv("WHEEL_TEST", "").lower() in {"true", "1", "yes"}, + reason="Test not supported in wheel.", ) diff --git a/tests/python/ci/test_ci.py b/tests/python/ci/test_ci.py index 27297e165fd6..0ed3f1701506 100644 --- a/tests/python/ci/test_ci.py +++ b/tests/python/ci/test_ci.py @@ -33,6 +33,7 @@ def parameterize_named(*values): return pytest.mark.parametrize(",".join(keys), [tuple(d.values()) for d in values]) +@tvm.testing.skip_if_wheel_test @pytest.mark.parametrize( "target_url,base_url,commit_sha,expected_url,expected_body", [ @@ -826,6 +827,7 @@ def run(source_type, data, check): ) +@tvm.testing.skip_if_wheel_test @parameterize_named( dict( tlcpackstaging_body={