I've made some changes to try to restrict the assumption that the number of partitions is 1 to two places:    * ContinuousCoalesceExec  * The output partitioner in ContinuousCoalesceRDD, since it's not obvious to me what the right strategy to get this would be in the general case. If you have ideas I'm open to removing this too.