mirror of
https://github.com/apache/arrow.git
synced 2026-10-29 00:56:31 +08:00
ARROW-4576: [Python] Fix error during benchmarks
This commit is contained in:
1 parent
148213e3be
commit
67e8069d1c
1 file changed
+1
-1
@@ -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)
|
||||
|
||||
Reference in new issue
Block a user