When trying to use joblib.dump to pickle the model, "pickle.PicklingError" was raised. Not sure how to fix it