Remove broken sort.

This wasn't the correct way to do it anyways (we should use the same
random sample method that Spark uses).