Here, you can use the choice argument of add_argument.