Skip to content

Commit f2302ca

Browse files
committed
chore(bigquery): explicitly set QueryResultsFormat.ARROW in queryWithArrowZeroCopy benchmark
1 parent 10e2c32 commit f2302ca

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

java-bigquery/benchmark/src/main/java/com.google.cloud.bigquery/QueryBenchmark.java

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,10 @@ public void queryWithArrowRowBased(QueryParams queryParams, Blackhole blackhole)
113113
public void queryWithArrowZeroCopy(QueryParams queryParams, Blackhole blackhole)
114114
throws Exception {
115115
QueryJobConfiguration config =
116-
QueryJobConfiguration.newBuilder(queryParams.queries).setUseLegacySql(false).build();
116+
QueryJobConfiguration.newBuilder(queryParams.queries)
117+
.setUseLegacySql(false)
118+
.setQueryResultsFormat(QueryResultsFormat.ARROW)
119+
.build();
117120
try (ArrowQueryResult result = bigquery.queryArrow(config)) {
118121
for (VectorSchemaRoot root : result) {
119122
blackhole.consume(root);

0 commit comments

Comments
 (0)