def load_midi_folder()

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

• 19 lines of code
• 4 McCabe index (conditional complexity)

def load_midi_folder(path):
    '''
    Load all midi files in specified folder
    
    Arguments:
    path : folder path containing midi files
    
    Return:
    midi: list of midi files in shape [768, 128]
    '''
    midi_list = []
    midi_name = []
    
    for f in glob.glob(path + '*.mid'):
        print(f)
        sample, name = load_midi(f)
        # Skip if input midi does not have at least 8 bars
        if not sample:
            print(f'File {f} does not have at least 8 bars! Exiting..')
            continue
        midi_list.append(sample[0])
        midi_name.append(name)
       
    return midi_list, midi_name