Fixed example and bench

jorgecarleitao · May 26, 2022 · b60c454 · b60c454
1 parent 1014a52
commit b60c454
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/benches/write_parquet.rs b/benches/write_parquet.rs
@@ -25,7 +25,7 @@ fn write(array: &dyn Array, encoding: Encoding) -> Result<()> {
         vec![Ok(columns)].into_iter(),
         &schema,
         options,
-        vec![encoding],
+        vec![vec![encoding]],
     )?;
 
     let writer = vec![];

diff --git a/examples/parquet_write.rs b/examples/parquet_write.rs
@@ -20,8 +20,12 @@ fn write_batch(path: &str, schema: Schema, columns: Chunk<Arc<dyn Array>>) -> Re
 
     let iter = vec![Ok(columns)];
 
-    let row_groups =
-        RowGroupIterator::try_new(iter.into_iter(), &schema, options, vec![Encoding::Plain])?;
+    let row_groups = RowGroupIterator::try_new(
+        iter.into_iter(),
+        &schema,
+        options,
+        vec![vec![Encoding::Plain]],
+    )?;
 
     // Create a new empty file
     let file = File::create(path)?;