I think this is handled in Settings -> Windows Update -> click the Advanced Options link -> turn on "Show a notification when your PC requires a reboot to finish updating"

I have only ever seen it ask me if I'm OK rebooting.