diff --git a/tools/distpartitioning/convert_partition.py b/tools/distpartitioning/convert_partition.py index de1e48e14e08..a719793997d1 100644 --- 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]