[CI] Skip some additional tests that are failing in the wheel (apache…

…#11969) This PR skips some additional tests that are failing in the nightly wheel.
driazati · Jun 30, 2022 · 288b983 · 288b983
1 parent 6a86c97
commit 288b983
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git 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
@@ -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={