Make predict function work for mixed logit with 1 draw

[timothyb0912]

Right now, calling predict when using a mixed logit model with num_draws=1 will cause an error. The following code snippet needs to included right after L1959:

if len(prob_array.shape) == 1:
    return prob_array