1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
# We need the absolute_import to avoid the local joblib to override the
# site one
from __future__ import absolute_import
import os as _os
import warnings as _warnings
# An environment variable to use the site joblib
if _os.environ.get('SKLEARN_SITE_JOBLIB', False):
with _warnings.catch_warnings():
_warnings.simplefilter("ignore")
# joblib imports may raise DeprecationWarning on certain Python
# versions
import joblib
from joblib import logger
from joblib import dump, load
from joblib import __version__
from joblib import effective_n_jobs
from joblib import hash
from joblib import cpu_count, Parallel, Memory, delayed
from joblib import parallel_backend, register_parallel_backend
else:
from ..externals import joblib
from ..externals.joblib import logger
from ..externals.joblib import dump, load
from ..externals.joblib import __version__
from ..externals.joblib import effective_n_jobs
from ..externals.joblib import hash
from ..externals.joblib import cpu_count, Parallel, Memory, delayed
from ..externals.joblib import parallel_backend, register_parallel_backend
__all__ = ["parallel_backend", "register_parallel_backend", "cpu_count",
"Parallel", "Memory", "delayed", "effective_n_jobs", "hash",
"logger", "dump", "load", "joblib", "__version__"]
|
