Merge "Make partition metadata write atomic" into main