8000 Export _joblib's register_parallel_backend · scikit-learn/scikit-learn@d5623a0 · GitHub
[go: up one dir, main page]
Skip to content

Commit d5623a0

Browse files
ogriselogrisel
committed
Export _joblib's register_parallel_backend
1 parent 4fc099d commit d5623a0

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

‎sklearn/ensemble/tests/test_forest.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,8 @@
2121

2222
import pytest
2323

24-
from sklearn.externals.joblib import parallel_backend
25-
from sklearn.externals.joblib import register_parallel_backend
24+
from sklearn.utils import parallel_backend
25+
from sklearn.utils import register_parallel_backend
2626
from sklearn.externals.joblib.parallel import LokyBackend
2727

2828
from sklearn.utils.testing import assert_almost_equal

‎sklearn/utils/__init__.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,8 @@
1616
check_symmetric)
1717
from .class_weight import compute_class_weight, compute_sample_weight
1818
from ._joblib import cpu_count, Parallel, Memory, delayed, hash
19-
from ._joblib import parallel_backend, effective_n_jobs
19+
from ._joblib import parallel_backend, register_parallel_backend
20+
from ._joblib import effective_n_jobs
2021
from ..exceptions import DataConversionWarning
2122
from ..utils.fixes import _Sequence as Sequence
2223
from .deprecation import deprecated
@@ -30,7 +31,7 @@
3031
"check_consistent_length", "check_X_y", 'indexable',
3132
"check_symmetric", "indices_to_mask", "deprecated",
3233
"cpu_count", "Parallel", "Memory", "delayed", "parallel_backend",
33-
"hash", "effective_n_jobs"]
34+
"register_parallel_backend", "hash", "effective_n_jobs"]
3435

3536
IS_PYPY = platform.python_implementation() == 'PyPy'
3637

0 commit comments

Comments
 (0)
0