From 3b8eabcf823e21e8e551ae1e6a7865934a0eb0ca Mon Sep 17 00:00:00 2001 From: Pierre Yger Date: Wed, 29 May 2024 15:33:32 +0200 Subject: [PATCH] WIP --- src/spikeinterface/preprocessing/whiten.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/spikeinterface/preprocessing/whiten.py b/src/spikeinterface/preprocessing/whiten.py index f3f0a1368b..2d56403e65 100644 --- a/src/spikeinterface/preprocessing/whiten.py +++ b/src/spikeinterface/preprocessing/whiten.py @@ -208,7 +208,11 @@ def compute_whitening_matrix( regularize_kwargs["assume_centered"] = True job_kwargs = get_global_job_kwargs() if "n_jobs" in job_kwargs and "n_jobs" not in regularize_kwargs: - regularize_kwargs["n_jobs"] = job_kwargs["n_jobs"] + n_jobs = job_kwargs["n_jobs"] + if isinstance(n_jobs, float) and 0 < n_jobs <= 1: + import os + n_jobs = int(n_jobs * os.cpu_count()) + regularize_kwargs["n_jobs"] = n_jobs estimator = sklearn.covariance.GraphicalLassoCV(**regularize_kwargs) estimator.fit(data) cov = estimator.covariance_