RemoveStageBarrier: revert previous changes to CoGroupedRDD