[BUG] n_jobs=-1 not working in spectral_connectivity_epochs (#177)

mne-tools · Mar 12, 2024 · 8833305 · 8833305
1 parent f9a2e65
commit 8833305
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mne_connectivity/spectral/epochs.py b/mne_connectivity/spectral/epochs.py
@@ -912,7 +912,7 @@ def spectral_connectivity_epochs(
     .. footbibliography::
     """
     if n_jobs != 1:
-        parallel, my_epoch_spectral_connectivity, _ = parallel_func(
+        parallel, my_epoch_spectral_connectivity, n_jobs = parallel_func(
             _epoch_spectral_connectivity, n_jobs, verbose=verbose
         )
 
@@ -1094,7 +1094,7 @@ def spectral_connectivity_epochs(
                     n_cons=n_cons, n_freqs=n_freqs, n_times=n_times_spectrum
                 )
                 if method[mtype_i] in _multivariate_methods:
-                    method_params.update(dict(n_signals=n_signals_use))
+                    method_params.update(dict(n_signals=n_signals_use, n_jobs=n_jobs))
                     if method[mtype_i] in _gc_methods:
                         method_params.update(dict(n_lags=gc_n_lags))
                 con_methods.append(mtype(**method_params))