[partition] Apply coding style

This applies the new(er) style to the whole module,
rather than just the files that recently changed.