From b3fcb290271d9e6a0eb219be3be3668d3f0b29b1 Mon Sep 17 00:00:00 2001 From: Connor Tsui Date: Tue, 30 Dec 2025 13:00:39 +1100 Subject: [PATCH] bring back lance compression benchmark Signed-off-by: Connor Tsui --- benchmarks/compress-bench/src/lib.rs | 3 ++- benchmarks/compress-bench/src/main.rs | 4 +++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/benchmarks/compress-bench/src/lib.rs b/benchmarks/compress-bench/src/lib.rs index 196a00b5ccf..34585ce88c7 100644 --- a/benchmarks/compress-bench/src/lib.rs +++ b/benchmarks/compress-bench/src/lib.rs @@ -7,7 +7,8 @@ use ::vortex::array::arrays::ChunkedArray; use ::vortex::array::arrays::recursive_list_from_list_view; use arrow_array::RecordBatch; use arrow_schema::Schema; - +#[cfg(feature = "lance")] +pub use lance_bench::compress::LanceCompressor; pub mod parquet; pub mod vortex; diff --git a/benchmarks/compress-bench/src/main.rs b/benchmarks/compress-bench/src/main.rs index 24233748e23..315a5b82f3b 100644 --- a/benchmarks/compress-bench/src/main.rs +++ b/benchmarks/compress-bench/src/main.rs @@ -5,6 +5,8 @@ use std::path::PathBuf; use std::time::Duration; use clap::Parser; +#[cfg(feature = "lance")] +use compress_bench::LanceCompressor; use compress_bench::parquet::ParquetCompressor; use compress_bench::vortex::VortexCompressor; use indicatif::ProgressBar; @@ -92,7 +94,7 @@ fn get_compressor(format: Format) -> Box { Format::OnDiskVortex => Box::new(VortexCompressor), Format::Parquet => Box::new(ParquetCompressor::new()), #[cfg(feature = "lance")] - Format::Lance => todo!(), + Format::Lance => Box::new(LanceCompressor), _ => unimplemented!("Compress bench not implemented for {format}"), } }