From 43b54a16be2bd32bf013a34b87b318db06922c30 Mon Sep 17 00:00:00 2001
From: Andrei Ivanov <andreii@nvidia.com>
Date: Wed, 25 Sep 2024 10:30:46 -0700
Subject: [PATCH] Fixing lint problem

---
 .../pytorch/graphbolt/impl/test_hetero_cached_feature.py      | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/python/pytorch/graphbolt/impl/test_hetero_cached_feature.py b/tests/python/pytorch/graphbolt/impl/test_hetero_cached_feature.py
index 0686e4c1cfcf..f3caf34da38e 100644
--- a/tests/python/pytorch/graphbolt/impl/test_hetero_cached_feature.py
+++ b/tests/python/pytorch/graphbolt/impl/test_hetero_cached_feature.py
@@ -16,8 +16,8 @@ def test_hetero_cached_feature(cached_feature_type):
     ):
         pytest.skip(
             "GPUCachedFeature tests are available only when testing the GPU backend."
-            if F._default_context_str != "gpu" else
-            "GPUCachedFeature requires a Volta or later generation NVIDIA GPU."
+            if F._default_context_str != "gpu"
+            else "GPUCachedFeature requires a Volta or later generation NVIDIA GPU."
         )
     device = F.ctx() if cached_feature_type == gb.gpu_cached_feature else None
     pin_memory = cached_feature_type == gb.gpu_cached_feature