From 7f339b9cd7919098af8aabe63521d52b7d04f319 Mon Sep 17 00:00:00 2001
From: Ubuntu <ubuntu@ip-172-31-52-174.us-west-2.compute.internal>
Date: Thu, 17 Oct 2024 01:13:29 +0000
Subject: [PATCH] fix convert_partition.py

---
 tools/distpartitioning/convert_partition.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tools/distpartitioning/convert_partition.py b/tools/distpartitioning/convert_partition.py
index 5013b6d40f20..b03053ccb6df 100644
--- a/tools/distpartitioning/convert_partition.py
+++ b/tools/distpartitioning/convert_partition.py
@@ -351,6 +351,8 @@ def _process_partition_gb(
         sorted_idx = (
             th.repeat_interleave(indptr[:-1], split_size, dim=0) + sorted_idx
         )
+    else:
+        sorted_idx = th.arange(len(edge_ids))
 
     return indptr, indices[sorted_idx], edge_ids[sorted_idx]