diff --git a/native/core/benches/parquet_read.rs b/native/core/benches/parquet_read.rs index a56c748e0f..52ab96eb1b 100644 --- a/native/core/benches/parquet_read.rs +++ b/native/core/benches/parquet_read.rs @@ -37,7 +37,6 @@ use comet::parquet::util::test_common::page_util::{ use perf::FlamegraphProfiler; use rand::{prelude::StdRng, RngExt, SeedableRng}; -use zstd::zstd_safe::WriteBuf; fn bench(c: &mut Criterion) { let expected_num_values: usize = NUM_PAGES * VALUES_PER_PAGE; @@ -177,7 +176,7 @@ impl TestColumnReader { fn load_page(&mut self) { if let Some(page) = self.pages.get_next_page().unwrap() { let num_values = page.num_values() as usize; - let buffer = Buffer::from_slice_ref(page.buffer().as_slice()); + let buffer = Buffer::from_slice_ref(page.buffer()); self.inner.set_page_v1(num_values, buffer, page.encoding()); } } diff --git a/native/core/src/parquet/util/test_common/page_util.rs b/native/core/src/parquet/util/test_common/page_util.rs index e4e3e30c45..6bda2db31a 100644 --- a/native/core/src/parquet/util/test_common/page_util.rs +++ b/native/core/src/parquet/util/test_common/page_util.rs @@ -33,7 +33,6 @@ use parquet::{ use super::random_numbers_range; use bytes::Bytes; -use zstd::zstd_safe::WriteBuf; pub trait DataPageBuilder { fn add_rep_levels(&mut self, max_level: i16, rep_levels: &[i16]); @@ -127,7 +126,7 @@ impl DataPageBuilder for DataPageBuilderImpl { let encoded_values = encoder .flush_buffer() .expect("consume_buffer() should be OK"); - self.buffer.extend_from_slice(encoded_values.as_slice()); + self.buffer.extend_from_slice(&encoded_values); } fn add_indices(&mut self, indices: Bytes) {