def parse_args()

in torchx/apps/utils/booth_main.py [0:0]

• 29 lines of code
• 1 McCabe index (conditional complexity)

def parse_args(argv: List[str]) -> argparse.Namespace:
    parser = argparse.ArgumentParser(
        description="evaluates the booth function at (x1, x2)"
    )
    parser.add_argument(
        "--x1",
        type=float,
        help="x1",
        required=True,
    )
    parser.add_argument(
        "--x2",
        type=float,
        help="x2",
        required=True,
    )
    parser.add_argument(
        "--tracker_base",
        type=str,
        help="URI of output directory to use as the tracker's base dir",
        default="/tmp/torchx-utils-booth",
    )
    parser.add_argument(
        "--trial_idx",
        type=int,
        help="trial index (ignore if not running hpo)",
        default=0,
    )
    return parser.parse_args(argv)