From 288b983b827d12539e23eda4fd9ec1ddd2c8cd1a Mon Sep 17 00:00:00 2001
From: Florin Blanaru <florin.blanaru96@gmail.com>
Date: Thu, 30 Jun 2022 20:50:21 +0100
Subject: [PATCH] [CI] Skip some additional tests that are failing in the wheel
 (#11969)

This PR skips some additional tests that are failing in the nightly wheel.
---
 python/tvm/testing/utils.py | 3 ++-
 tests/python/ci/test_ci.py  | 2 ++
 2 files changed, 4 insertions(+), 1 deletion(-)

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={