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

apache · Jun 30, 2022 · 6358b50 · 6358b50
1 parent 1aec41a
commit 6358b50
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/python/tvm/testing/utils.py b/python/tvm/testing/utils.py
@@ -95,7 +95,7 @@ def test_something():
 SKIP_SLOW_TESTS = os.getenv("SKIP_SLOW_TESTS", "").lower() in {"true", "1", "yes"}
 
 skip_if_wheel_test = pytest.mark.skipif(
-    os.getenv("WHEEL_TEST") is not None, reason="Test not supported in wheel."
+    os.getenv("WHEEL_TEST") is not None and len(os.getenv("WHEEL_TEST")) != 0, 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={