change convert_partition.py

dmlc · Oct 15, 2024 · 41a6b4a · 41a6b4a
1 parent 94030de
commit 41a6b4a
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/tools/distpartitioning/convert_partition.py b/tools/distpartitioning/convert_partition.py
@@ -352,8 +352,6 @@ def _process_partition_gb(
         sorted_idx = (
             th.repeat_interleave(indptr[:-1], split_size, dim=0) + sorted_idx
         )
-    else:
-        sorted_idxs=th.arange(len(edge_ids))
 
     return indptr, indices[sorted_idx], edge_ids[sorted_idx]