for udfs with default values we need to b able to remove columns after the prediction is done.
for udfs with default values we need to b able to remove columns after the prediction is done.