From beab715313d6fda8efe76245316966cce8135066 Mon Sep 17 00:00:00 2001 From: Ubuntu Date: Mon, 14 Oct 2024 03:26:11 +0000 Subject: [PATCH] change format --- python/dgl/distributed/partition.py | 2 +- tools/distpartitioning/data_shuffle.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/python/dgl/distributed/partition.py b/python/dgl/distributed/partition.py index 42ace8a7f4d7..0bf6deaaba49 100644 --- a/python/dgl/distributed/partition.py +++ b/python/dgl/distributed/partition.py @@ -1270,7 +1270,7 @@ def get_homogeneous(g, balance_ntypes): ) val = np.cumsum( val - ).tolist()# note computing the cumulative sum of array elements. + ).tolist() # note computing the cumulative sum of array elements. assert val[-1] == g.num_nodes(ntype) for etype in g.canonical_etypes: etype_id = g.get_etype_id(etype) diff --git a/tools/distpartitioning/data_shuffle.py b/tools/distpartitioning/data_shuffle.py index 6aba4fb59b3a..daae0fcbb3fd 100644 --- a/tools/distpartitioning/data_shuffle.py +++ b/tools/distpartitioning/data_shuffle.py @@ -1411,7 +1411,7 @@ def prepare_local_data(src_data, local_part_id): for rank_id, last_id in enumerate(gather_list): prev_last_ids[ rank_id + (local_part_id * world_size) - ] = last_id.item() + ] = last_id.item() if rank == 0: # get meta-data from all partitions and merge them on rank-0