It looks like you're just doing a `map` here, and producing a list. Please use `joblib.parallel.Parallel`. There are numerous examples of its use throughout scikit-learn.