diff --git a/arrow-avro/src/reader/async_reader/mod.rs b/arrow-avro/src/reader/async_reader/mod.rs index c034411edb03..31df7cdf110b 100644 --- a/arrow-avro/src/reader/async_reader/mod.rs +++ b/arrow-avro/src/reader/async_reader/mod.rs @@ -228,7 +228,7 @@ impl AsyncAvroFileReader { /// Drain any remaining buffered records from the decoder. #[inline] fn poll_flush(&mut self) -> Poll>> { - match self.decoder.flush() { + match self.decoder.flush_block() { Ok(Some(batch)) => { self.reader_state = ReaderState::Flushing; Poll::Ready(Some(Ok(batch))) @@ -512,7 +512,7 @@ impl AsyncAvroFileReader { // We have a full batch ready, emit it // (This is not mutually exclusive with the block being finished, so the state change is valid) if self.decoder.batch_is_full() { - return match self.decoder.flush() { + return match self.decoder.flush_block() { Ok(Some(batch)) => Poll::Ready(Some(Ok(batch))), Ok(None) => self.finish_with_error(AvroError::General( "Decoder reported a full batch, but flush returned None".into(),