diff --git a/vortex-serde/src/layouts/read/recordbatchreader.rs b/vortex-serde/src/layouts/read/recordbatchreader.rs index 6612acc367..f59120a775 100644 --- a/vortex-serde/src/layouts/read/recordbatchreader.rs +++ b/vortex-serde/src/layouts/read/recordbatchreader.rs @@ -20,12 +20,12 @@ fn vortex_to_arrow(result: VortexResult) -> Result { +pub struct VortexRecordBatchReader { stream: LayoutBatchStream, arrow_schema: SchemaRef, } -impl VortexRecordBatchReader { +impl VortexRecordBatchReader { pub fn new(stream: LayoutBatchStream) -> VortexResult> { let arrow_schema = Arc::new(infer_schema(stream.schema().dtype())?); Ok(VortexRecordBatchReader { @@ -35,7 +35,7 @@ impl VortexRecordBatchReader { } } -impl Iterator for VortexRecordBatchReader { +impl Iterator for VortexRecordBatchReader { type Item = Result; fn next(&mut self) -> Option { @@ -44,7 +44,7 @@ impl Iterator for VortexRecordBatchRea } } -impl RecordBatchReader for VortexRecordBatchReader { +impl RecordBatchReader for VortexRecordBatchReader { fn schema(&self) -> SchemaRef { self.arrow_schema.clone() }