def load_midi()

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

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

def load_midi(file):
    '''
    Load single midi file
    
    Return:
    List containting a single midi in shape [768, 128]
    '''
    try:
        midi = piano.read(file)
    except ValueError:
        print("Midi file does not exist")
        
    # Binarize merged track
    merged = midi.blend(mode='sum')
    binary = np.array(merged > 0).astype(int)
    # Sample beginning of the 8 bars if input too long
    sample = sample_midi(binary)
    
    return sample, file.split('/')[-1]