Since the distribution file contains the logic for the restart, admins can distribute the entire munki package for upgrades and use a ChoiceChangesXML.
For DEP configurations, if you customize your install in anyway, you must extract the subpackages and create your own distribution package (since you can't have distribution package inside of a distribution package). You can re-insert the logic in the new distribution package if needed.
Greg Neagle