diff --git a/benches/arrow_reader.rs b/benches/arrow_reader.rs index f466188..10f2899 100644 --- a/benches/arrow_reader.rs +++ b/benches/arrow_reader.rs @@ -18,8 +18,8 @@ use std::fs::File; use criterion::{criterion_group, criterion_main, Criterion}; -use datafusion_orc::arrow_reader::ArrowReaderBuilder; use futures_util::TryStreamExt; +use orc_rust::arrow_reader::ArrowReaderBuilder; fn basic_path(path: &str) -> String { let dir = env!("CARGO_MANIFEST_DIR"); diff --git a/examples/datafusion_integration.rs b/examples/datafusion_integration.rs index b9df8c7..e75c6c1 100644 --- a/examples/datafusion_integration.rs +++ b/examples/datafusion_integration.rs @@ -21,8 +21,8 @@ use datafusion::physical_plan::{ }; use datafusion::prelude::*; use datafusion_expr::{Expr, LogicalPlanBuilder}; -use datafusion_orc::ArrowReaderBuilder; use datafusion_physical_expr::PhysicalSortExpr; +use orc_rust::ArrowReaderBuilder; use async_trait::async_trait; use futures_util::StreamExt; diff --git a/src/bin/orc-metadata.rs b/src/bin/orc-metadata.rs index 5c0889f..4524df4 100644 --- a/src/bin/orc-metadata.rs +++ b/src/bin/orc-metadata.rs @@ -1,7 +1,7 @@ use std::{error::Error, fs::File, path::PathBuf, sync::Arc}; use clap::Parser; -use datafusion_orc::{reader::metadata::read_metadata, stripe::Stripe}; +use orc_rust::{reader::metadata::read_metadata, stripe::Stripe}; #[derive(Parser)] #[command(version, about, long_about = None)] diff --git a/src/lib.rs b/src/lib.rs index 036d3aa..511eabc 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -5,7 +5,7 @@ //! //! ```no_run //! # use std::fs::File; -//! # use datafusion_orc::arrow_reader::{ArrowReader, ArrowReaderBuilder}; +//! # use orc_rust::arrow_reader::{ArrowReader, ArrowReaderBuilder}; //! let file = File::open("/path/to/file.orc").unwrap(); //! let reader = ArrowReaderBuilder::try_new(file).unwrap().build(); //! let record_batches = reader.collect::, _>>().unwrap(); diff --git a/tests/basic/main.rs b/tests/basic/main.rs index eb7d1c9..1860335 100644 --- a/tests/basic/main.rs +++ b/tests/basic/main.rs @@ -4,12 +4,12 @@ use std::sync::Arc; use arrow::datatypes::{DataType, Field, Schema, TimeUnit}; use arrow::record_batch::{RecordBatch, RecordBatchReader}; use arrow::util::pretty; -use datafusion_orc::arrow_reader::{ArrowReader, ArrowReaderBuilder}; -#[cfg(feature = "async")] -use datafusion_orc::async_arrow_reader::ArrowStreamReader; -use datafusion_orc::projection::ProjectionMask; #[cfg(feature = "async")] use futures_util::TryStreamExt; +use orc_rust::arrow_reader::{ArrowReader, ArrowReaderBuilder}; +#[cfg(feature = "async")] +use orc_rust::async_arrow_reader::ArrowStreamReader; +use orc_rust::projection::ProjectionMask; use crate::misc::{LONG_BOOL_EXPECTED, LONG_STRING_DICT_EXPECTED, LONG_STRING_EXPECTED}; diff --git a/tests/integration/main.rs b/tests/integration/main.rs index b38bff7..aecee91 100644 --- a/tests/integration/main.rs +++ b/tests/integration/main.rs @@ -9,7 +9,7 @@ use arrow::{ }; use pretty_assertions::assert_eq; -use datafusion_orc::arrow_reader::ArrowReaderBuilder; +use orc_rust::arrow_reader::ArrowReaderBuilder; fn read_orc_file(name: &str) -> RecordBatch { let path = format!(