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