It would be nice to see how much of a performance penalty we would pay for doing the sqrt for every edge. If it's significant, it would be better to do it once in an init method on PartitionStrategy.