diff --git a/arro3-io/src/ipc.rs b/arro3-io/src/ipc.rs index 93cbc4b..cb1640a 100644 --- a/arro3-io/src/ipc.rs +++ b/arro3-io/src/ipc.rs @@ -33,6 +33,7 @@ pub fn write_ipc(data: AnyRecordBatch, file: FileWriter) -> PyArrowResult<()> { for batch in reader { writer.write(&batch?)?; } + writer.finish()?; Ok(()) } @@ -45,5 +46,6 @@ pub fn write_ipc_stream(data: AnyRecordBatch, file: FileWriter) -> PyArrowResult for batch in reader { writer.write(&batch?)?; } + writer.finish()?; Ok(()) } diff --git a/arro3-io/src/json.rs b/arro3-io/src/json.rs index 852c277..4128850 100644 --- a/arro3-io/src/json.rs +++ b/arro3-io/src/json.rs @@ -102,5 +102,6 @@ pub fn write_ndjson( for batch in data.into_reader()? { writer.write(&batch?)?; } + writer.finish()?; Ok(()) } diff --git a/arro3-io/src/parquet.rs b/arro3-io/src/parquet.rs index ccd9a2c..10f3d3f 100644 --- a/arro3-io/src/parquet.rs +++ b/arro3-io/src/parquet.rs @@ -213,5 +213,6 @@ pub(crate) fn write_parquet( for batch in reader { writer.write(&batch?).unwrap(); } + writer.close().unwrap(); Ok(()) }