linting

dmlc · Aug 29, 2024 · 10fb015 · 10fb015
1 parent 97f232b
commit 10fb015
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/python/pytorch/graphbolt/test_utils.py b/tests/python/pytorch/graphbolt/test_utils.py
@@ -142,7 +142,9 @@ def test_exclude_seed_edges_gpu(use_datapipe, async_op):
     if use_datapipe:
         datapipe = datapipe.exclude_seed_edges(asynchronous=async_op)
     else:
-        datapipe = datapipe.transform(partial(gb.exclude_seed_edges, async_op=async_op))
+        datapipe = datapipe.transform(
+            partial(gb.exclude_seed_edges, async_op=async_op)
+        )
     if torch.cuda.get_device_capability()[0] < 7:
         original_row_node_ids = [
             torch.tensor([0, 3, 4, 2, 5, 7]).to(F.ctx()),