Correcting the misleading reason for skipping the test. (#7749)

dmlc · Aug 30, 2024 · bb02829 · bb02829
1 parent a719465
commit bb02829
Showing 1 changed file with 3 additions and 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
@@ -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",