fix

viirya · Apr 16, 2024 · e4e0f82 · e4e0f82
1 parent 83d5782
commit e4e0f82
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/datafusion/physical-plan/src/sorts/sort.rs b/datafusion/physical-plan/src/sorts/sort.rs
@@ -605,7 +605,7 @@ pub(crate) fn sort_batch(
         .map(|c| take(c.as_ref(), &indices, None))
         .collect::<Result<_, _>>()?;
 
-    let options = RecordBatchOptions::new().with_row_count(Some(batch.num_rows()));
+    let options = RecordBatchOptions::new().with_row_count(Some(indices.len()));
     Ok(RecordBatch::try_new_with_options(
         batch.schema(),
         columns,