Don't mock machine learning models in unit tests