Reboot options when installing SEP client

Is there a way to get a more custimized reboot option when installing the client?  Ideally I'd like the user to be prompted with the same settings I have configured in the "Restart Options" under "General Settings" in the SEPM.  When I client installs I'd like the user to only have the option to WAIT or REBOOT, no option to DO NOT RESTART.