sr: updater: Fix multi-stage docs and code

* Fix reboot_now comments to reflect actual usage.

* Fix get_stage arg check, it only requires one arg (but allow two
  for compatibility).

Change-Id: I692d626185af24620382f7b318c2b77f643ccfb8