update plot helper to pass-through output

trevorstephens · trevorstephens · commit e8c8a0cad2ad · 2017-10-29T09:45:45.000+11:00
diff --git a/sklearn/partial_dependence.py b/sklearn/partial_dependence.py
@@ -531,7 +531,8 @@ def convert_feature(fx):
 
     # compute PD functions
     pd_result = Parallel(n_jobs=n_jobs, verbose=verbose)(
-        delayed(partial_dependence)(est, fxs, X=X, method=method,
+        delayed(partial_dependence)(est, fxs, X=X, output=output,
+                                    method=method,
                                     grid_resolution=grid_resolution,
                                     percentiles=percentiles)
         for fxs in features)
