List value not in prediction class leads to error

[ethompson93]

I get the attached error message when there are list values that are not present in the prediction class:

data['pred_class'] = [[1] * 5]
data['list_values'] = [0,1]

error_message.txt