Skip to content

Commit 61ca279

Browse files
committed
linting
1 parent c3e8e89 commit 61ca279

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/python/pytorch/graphbolt/impl/test_cooperative_minibatching_utils.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,9 @@ def test_rank_sort_and_unique_and_compact(dtype, rank):
4141
res3 = torch.ops.graphbolt.rank_sort(nodes_list1[::-1], rank, WORLD_SIZE)
4242

4343
# This function is deterministic. Call with identical arguments and check.
44-
for (nodes1, idx1, offsets1), (nodes3, idx3, offsets3) in zip(res1, reversed(res3)):
44+
for (nodes1, idx1, offsets1), (nodes3, idx3, offsets3) in zip(
45+
res1, reversed(res3)
46+
):
4547
assert_equal(nodes1, nodes3)
4648
assert_equal(idx1, idx3)
4749
assert_equal(offsets1.diff(), offsets3.diff())

0 commit comments

Comments
 (0)