def get_midi_paths()

in transformer-xl/utils/music_encoder.py [0:0]

• 7 lines of code
• 5 McCabe index (conditional complexity)

def get_midi_paths(dataset_dir):
    if not os.path.exists(dataset_dir):
        raise ValueError('Cannot find dataset. Please provide the correct path')
        
    train_paths = [os.path.join(dataset_dir,'train', midi) for midi in os.listdir(os.path.join(dataset_dir,'train'))]
    validation_paths = [os.path.join(dataset_dir,'valid', midi) for midi in os.listdir(os.path.join(dataset_dir,'valid'))]
    test_paths = [os.path.join(dataset_dir,'test', midi) for midi in os.listdir(os.path.join(dataset_dir,'test'))]

    return train_paths, validation_paths, test_paths