def METHOD_NAME(parser):
    parser.add_argument(
        "--client_num_per_round",
        type=int,
        default=3,
        metavar="NN",
        help="number of workers",
    )
    parser.add_argument(
        "--comm_round",
        type=int,
        default=10,
        help="how many round of communications we should use",
    )
    args = parser.parse_args()
    return args