Skip to content

Commit 2aa2e5f

Browse files
fix: compatibility with datafusion
1 parent cc1ef1f commit 2aa2e5f

File tree

1 file changed

+5
-10
lines changed

1 file changed

+5
-10
lines changed

crates/iceberg/src/writer/base_writer/equality_delete_writer.rs

Lines changed: 5 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -232,7 +232,6 @@ impl FieldProjector {
232232
mod test {
233233
use arrow_select::concat::concat_batches;
234234
use bytes::Bytes;
235-
use futures::AsyncReadExt;
236235
use itertools::Itertools;
237236
use std::{collections::HashMap, sync::Arc};
238237

@@ -266,15 +265,11 @@ mod test {
266265
assert_eq!(data_file.file_format, DataFileFormat::Parquet);
267266

268267
// read the written file
269-
let mut input_file = file_io
270-
.new_input(data_file.file_path.clone())
271-
.unwrap()
272-
.reader()
273-
.await
274-
.unwrap();
275-
let mut res = vec![];
276-
let file_size = input_file.read_to_end(&mut res).await.unwrap();
277-
let reader_builder = ParquetRecordBatchReaderBuilder::try_new(Bytes::from(res)).unwrap();
268+
let input_file = file_io.new_input(data_file.file_path.clone()).unwrap();
269+
// read the written file
270+
let input_content = input_file.read().await.unwrap();
271+
let reader_builder =
272+
ParquetRecordBatchReaderBuilder::try_new(input_content.clone()).unwrap();
278273
let metadata = reader_builder.metadata().clone();
279274

280275
// check data

0 commit comments

Comments
 (0)