diff --git a/python/benchmarks/streaming.py b/python/benchmarks/streaming.py index 2012141a9e3..c0c63e6ef4c 100644 --- a/python/benchmarks/streaming.py +++ b/python/benchmarks/streaming.py @@ -62,7 +62,7 @@ def setup(self, chunk_size): stream_writer = pa.RecordBatchStreamWriter(sink, schema) for batch in batches: stream_writer.write_batch(batch) - self.source = sink.get_result() + self.source = sink.getvalue() def time_read_to_dataframe(self, *args): reader = pa.RecordBatchStreamReader(self.source)