Make BenchmarkSelectiveStreamReaders load LazyBlock
introduced
LazyBlock in OrcSelectiveRecordReader.
Change ensures that the block is loaded for benchmarking.
@@ -138,7 +138,7 @@ public class BenchmarkSelectiveStreamReaders
}
if (page.getPositionCount() > 0) {
- blocks.add(page.getBlock(0));
+ blocks.add(page.getBlock(0).getLoadedBlock());
}
}
return blocks;
