ENH try memoising partial models in pipelines