B Z@s.dZddlmZddlmZmZdddZdS) aParallel utility function using joblib copied from https://github.com/mne-tools/mne-python Author: Alexandre Gramfort License: Simplified BSD changes for statsmodels (Josef Perktold) - try import from joblib directly, (doesn't import all of sklearn) )print_function)ModuleUnavailableWarningmodule_unavailable_docc Csyyddlm}m}Wn$tk r:ddlm}m}YnX|||d}||}|dkryddl}|}Wn6ttfk rddl}| t dt d}YnXWn:tk rddl}| t dt d}|}t }YnX|||fS) aReturn parallel instance with delayed function Util function to use joblib only if available Parameters ---------- func: callable A function n_jobs: int Number of jobs to run in parallel verbose: int Verbosity level Returns ------- parallel: instance of joblib.Parallel or list The parallel object my_func: callable func if not parallel or delayed(func) n_jobs: int Number of jobs >= 0 Examples -------- >>> from math import sqrt >>> from statsmodels.tools.parallel import parallel_func >>> parallel, p_func, n_jobs = parallel_func(sqrt, n_jobs=-1, verbose=0) >>> print(n_jobs) >>> parallel(p_func(i**2) for i in range(10)) r)Paralleldelayed)verboseNmultiprocessingjoblib)r rr ImportErrorZsklearn.externals.joblibr cpu_countNotImplementedErrorwarningswarnrformatrlist) funcZn_jobsrrrZparallelZmy_funcr rr9lib/python3.7/site-packages/statsmodels/tools/parallel.py parallel_funcs0     rN)r)__doc__Z __future__rZstatsmodels.tools.sm_exceptionsrrrrrrr s