| 298 | } |
| 299 | |
| 300 | Result<std::shared_ptr<::arrow::Table>> Table(int64_t rows_per_chunk, |
| 301 | int num_chunks = 1) override { |
| 302 | ARROW_ASSIGN_OR_RAISE(RecordBatchVector batches, |
| 303 | RecordBatches(rows_per_chunk, num_chunks)); |
| 304 | return ::arrow::Table::FromRecordBatches(batches); |
| 305 | } |
| 306 | |
| 307 | std::shared_ptr<::arrow::Schema> Schema() override { return schema_; } |
| 308 |