ARROW-4576: [Python] Fix error during benchmarks

This commit is contained in:
Antoine Pitrou committed 2019-02-14 19:09:21 +01:00
1 parent 148213e3be
commit 67e8069d1c
1 file changed
+1 -1
+1 -1
View File
@@ -62,7 +62,7 @@ class StreamReader(object):
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)