change format

dmlc · Oct 14, 2024 · beab715 · beab715
1 parent 0e9cf4c
commit beab715
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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