From bb02829b6aa3b850e92dfd021df4b10692d5c1f6 Mon Sep 17 00:00:00 2001
From: Andrei Ivanov <32910461+drivanov@users.noreply.github.com>
Date: Fri, 30 Aug 2024 10:18:16 -0700
Subject: [PATCH] Correcting the misleading reason for skipping the test.
 (#7749)

---
 tests/python/pytorch/graphbolt/impl/test_gpu_graph_cache.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/python/pytorch/graphbolt/impl/test_gpu_graph_cache.py b/tests/python/pytorch/graphbolt/impl/test_gpu_graph_cache.py
index e6034cf77019..6c6d242f14c1 100644
--- a/tests/python/pytorch/graphbolt/impl/test_gpu_graph_cache.py
+++ b/tests/python/pytorch/graphbolt/impl/test_gpu_graph_cache.py
@@ -11,7 +11,9 @@
 @unittest.skipIf(
     F._default_context_str != "gpu"
     or torch.cuda.get_device_capability()[0] < 7,
-    reason="GPUCachedFeature requires a Volta or later generation NVIDIA GPU.",
+    reason="GPUCachedFeature tests are available only on GPU."
+    if F._default_context_str != "gpu"
+    else "GPUCachedFeature requires a Volta or later generation NVIDIA GPU.",
 )
 @pytest.mark.parametrize(
     "indptr_dtype",