Pass Distribution_Type to Predict and handle Distribution_Type acoordingly.