Actually this is slightly more complicated. We want to make sure that we use the same partitioning strategy even when repartitioning the edge table (e.g. in `subgraph()`). Looking into the best way to do this now.