joblib can't pickle function as arguments