Switch to using joblib.dump and joblib.load for serialization.