fix arrow imports

kernel/src/log_segment/tests.rs

@@ -2,7 +2,6 @@ use std::{path::PathBuf, sync::Arc};
 
 use itertools::Itertools;
 use object_store::{memory::InMemory, path::Path, ObjectStore};
-use parquet::arrow::ArrowWriter;
 use url::Url;
 
 use crate::actions::visitors::AddVisitor;
@@ -16,6 +15,7 @@ use crate::engine::default::filesystem::ObjectStoreFileSystemClient;
 use crate::engine::default::DefaultEngine;
 use crate::engine::sync::SyncEngine;
 use crate::log_segment::LogSegment;
+use crate::parquet::arrow::ArrowWriter;
 use crate::path::ParsedLogPath;
 use crate::scan::test_utils::{
     add_batch_simple, add_batch_with_remove, sidecar_batch_with_given_paths,

kernel/src/scan/mod.rs

@@ -663,11 +663,10 @@ pub fn selection_vector(
 // some utils that are used in file_stream.rs and state.rs tests
 #[cfg(test)]
 pub(crate) mod test_utils {
-    use std::sync::Arc;
-
-    use arrow_array::{RecordBatch, StringArray};
-    use arrow_schema::{DataType, Field, Schema as ArrowSchema};
+    use crate::arrow::array::{RecordBatch, StringArray};
+    use crate::arrow::datatypes::{DataType, Field, Schema as ArrowSchema};
     use itertools::Itertools;
+    use std::sync::Arc;
 
     use crate::{
         actions::get_log_schema,

kernel/src/utils.rs

@@ -13,7 +13,7 @@ pub(crate) use require;
 
 #[cfg(test)]
 pub(crate) mod test_utils {
-    use arrow_array::RecordBatch;
+    use crate::arrow::array::RecordBatch;
     use itertools::Itertools;
     use object_store::local::LocalFileSystem;
     use object_store::ObjectStore;