Robust Learning of Diverse Code Edits