favor replace

Eventual-Inc · Feb 14, 2024 · e1106d6 · e1106d6
1 parent dbe9fd7
commit e1106d6
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/daft-scan/src/scan_task_iters.rs b/src/daft-scan/src/scan_task_iters.rs
@@ -98,9 +98,7 @@ impl Iterator for MergeByFileSize {
             };
 
             if next_item.size_bytes().is_none() || !self.can_merge(&next_item) {
-                let curr_acc = self.accumulator.take().map(Ok);
-                self.accumulator = Some(next_item);
-                return curr_acc;
+                return self.accumulator.replace(next_item).map(Ok);
             }
 
             self.accumulator = Some(Arc::new(