Please do pull in my change - that will make this faster by building directly a columnar representation of EdgePartition.