Merge branch 'dev-2.0.0-rc' into dev-2.0.0-sbt-dev

FederatedAI · Oct 11, 2023 · bf6edba · bf6edba
2 parents 6750d92 + 570b2e1
commit bf6edba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/fate/arch/dataframe/ops/_sort.py b/python/fate/arch/dataframe/ops/_sort.py
@@ -105,7 +105,7 @@ def _extract_columns(r_id):
     blocks_with_id = []
 
     for block_id in range(0, n, block_row_size):
-        block_indexes = indexes[block_id * block_row_size : (block_id + 1) * block_row_size]
+        block_indexes = indexes[block_id * block_row_size : min(n, (block_id + 1) * block_row_size)]
 
         blocks = [[] for _ in range(block_num)]