Fixes failing import _joblib_parallel_args

sdpython · sdpython · commit 75a83171f33e · 2022-02-18T15:10:35.000+01:00
diff --git a/mlinsights/mlmodel/interval_regressor.py b/mlinsights/mlmodel/interval_regressor.py
@@ -6,7 +6,6 @@
 import numpy.random
 from sklearn.base import RegressorMixin, clone, BaseEstimator
 from sklearn.utils._joblib import Parallel, delayed
-from sklearn.utils.fixes import _joblib_parallel_args
 try:
     from tqdm import tqdm
 except ImportError:  # pragma: no cover
@@ -93,7 +92,7 @@ def _fit_piecewise_estimator(i, est, X, y, sample_weight, alpha):
 
         self.estimators_ = \
             Parallel(n_jobs=self.n_jobs, verbose=verbose,
-                     **_joblib_parallel_args(prefer='threads'))(
+                     prefer='threads')(
                 delayed(_fit_piecewise_estimator)(
                     i, estimators[i], X, y, sample_weight, self.alpha)
                 for i in loop)
