Remove from substrait

Chen-Yuan-Lai · Feb 5, 2025 · f476a64 · f476a64
1 parent 920daa8
commit f476a64
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/datafusion/substrait/Cargo.toml b/datafusion/substrait/Cargo.toml
@@ -31,7 +31,6 @@ rust-version = { workspace = true }
 workspace = true
 
 [dependencies]
-arrow-buffer = { workspace = true }
 async-recursion = "1.0"
 async-trait = { workspace = true }
 chrono = { workspace = true }

diff --git a/datafusion/substrait/src/logical_plan/consumer.rs b/datafusion/substrait/src/logical_plan/consumer.rs
@@ -15,7 +15,8 @@
 // specific language governing permissions and limitations
 // under the License.
 
-use arrow_buffer::{IntervalDayTime, IntervalMonthDayNano, OffsetBuffer};
+use arrow::array::types::{IntervalDayTime, IntervalMonthDayNano};
+use arrow::buffer::OffsetBuffer;
 use async_recursion::async_recursion;
 use datafusion::arrow::array::MapArray;
 use datafusion::arrow::datatypes::{
@@ -67,7 +68,7 @@ use datafusion::logical_expr::{
 };
 use datafusion::prelude::{lit, JoinType};
 use datafusion::{
-    error::Result, logical_expr::utils::split_conjunction, prelude::Column,
+    arrow, error::Result, logical_expr::utils::split_conjunction, prelude::Column,
     scalar::ScalarValue,
 };
 use std::collections::HashSet;
@@ -3278,7 +3279,8 @@ mod test {
         from_substrait_literal_without_names, from_substrait_rex,
         DefaultSubstraitConsumer,
     };
-    use arrow_buffer::IntervalMonthDayNano;
+    use arrow::array::types::IntervalMonthDayNano;
+    use datafusion::arrow;
     use datafusion::common::DFSchema;
     use datafusion::error::Result;
     use datafusion::execution::SessionState;

diff --git a/datafusion/substrait/src/logical_plan/producer.rs b/datafusion/substrait/src/logical_plan/producer.rs
@@ -2535,7 +2535,8 @@ mod test {
         from_substrait_named_struct, from_substrait_type_without_names,
         DefaultSubstraitConsumer,
     };
-    use arrow_buffer::{IntervalDayTime, IntervalMonthDayNano};
+    use arrow::array::types::{IntervalDayTime, IntervalMonthDayNano};
+    use datafusion::arrow;
     use datafusion::arrow::array::{
         GenericListArray, Int64Builder, MapBuilder, StringBuilder,
     };