def parse_args()

in dataflux_pytorch/benchmark/checkpointing/singlenode/train.py [0:0]

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

def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument("--project", type=str)
    parser.add_argument("--ckpt-dir-path", type=str)
    parser.add_argument("--save-only-latest",
                        action="store_true",
                        default=False)
    parser.add_argument("--layers", type=int, default=100)
    parser.add_argument("--steps", type=int, default=5)
    parser.add_argument("--disable-multipart",
                        action="store_true",
                        default=False)
    parser.add_argument("--clear-kernel-cache",
                        action="store_true",
                        default=False)
    parser.add_argument(
        '--checkpoint',
        choices=[DF_LIGHTNING, ASYNC_DF_LIGHTNING, NO_DF, NO_LIGHTNING],
        default=DF_LIGHTNING)
    return parser.parse_args()