WIP

douglas-raillard-arm · Nov 9, 2023 · 61e5db7 · 61e5db7
1 parent 0552231
commit 61e5db7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/tools/analysis/trace-tools/Cargo.toml b/tools/analysis/trace-tools/Cargo.toml
@@ -15,6 +15,8 @@ traceevent = { path = "../traceevent" }
 thiserror = "1.0"
 smartstring = "1.0"
 arrow2 = { version = "0.18.0", features = ["io_parquet", "io_parquet_compression"] }
+polars-arrow = { version = "0.34", features = ["io_parquet", "io_parquet_compression"] }
+polars-error = "0.34"
 crossbeam = "0.8"
 
 [target.'cfg(target_arch = "x86_64")'.dependencies]

diff --git a/tools/analysis/trace-tools/src/lib/parquet.rs b/tools/analysis/trace-tools/src/lib/parquet.rs
@@ -12,7 +12,9 @@ use std::{
     sync::Arc,
 };
 
-use arrow2::{
+// use arrow2::error::Error as ArrowError;
+// use arrow2::{
+use polars_arrow::{
     array::MutablePrimitiveArray,
     // TODO: remove wildcard import
     array::*,
@@ -21,8 +23,9 @@ use arrow2::{
     datatypes::{DataType, Field, Schema},
     // TODO: remove wildcard import
     io::parquet::write::*,
-    error::Error as ArrowError,
 };
+use polars_error::PolarsError as ArrowError;
+
 use crossbeam::{
     channel::{bounded, Sender},
     thread::{scope, Scope, ScopedJoinHandle},
@@ -602,7 +605,7 @@ impl EventWriteState {
     }
 }
 
-fn main2() -> Result<(), arrow2::error::Error> {
+fn main2() -> Result<(), ArrowError> {
     // declare arrays
     let a = Int8Array::from(&[Some(1), None, Some(3)]);
     let b = Int32Array::from(&[Some(2), None, Some(6)]);