def parse_args()

in ai-judge/inference.py [0:0]

• 10 lines of code
• 3 McCabe index (conditional complexity)

def parse_args():
    parser = argparse.ArgumentParser()
    group = parser.add_mutually_exclusive_group(required=True)
    group.add_argument('--midi_file', type=str, dest='midi_file')
    group.add_argument('--midi_folder', type=str, dest='midi_folder')
    parser.add_argument('--save_path', type=str, dest='save_path')
    args = parser.parse_args()

    # test arguments
    if not args.midi_folder and not args.midi_file:
        assert 0, print("Please specify the midi file or a folder of midi files")

    return args