Adjust n_jobs

tsbinns · Aug 6, 2024 · 2a9c1e2 · 2a9c1e2
1 parent 30dc4e7
commit 2a9c1e2
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/examples/surrogate_connectivity.py b/examples/surrogate_connectivity.py
@@ -13,17 +13,13 @@
 
 # %%
 
-from multiprocessing import cpu_count
-
 import matplotlib.pyplot as plt
 import mne
 import numpy as np
 from mne.datasets import somato
 
 from mne_connectivity import make_surrogate_data, spectral_connectivity_epochs
 
-n_jobs = cpu_count() // 2
-
 ########################################################################################
 # Background
 # ----------
@@ -126,7 +122,7 @@
     print(f"Computing connectivity for shuffle {shuffle_i+1} of {n_shuffles}")
     surrogate_con.append(
         spectral_connectivity_epochs(
-            surrogate, method="imcoh", indices=indices, n_jobs=n_jobs, verbose=False
+            surrogate, method="imcoh", indices=indices, verbose=False
         )
     )
 
@@ -286,7 +282,7 @@
     print(f"Computing connectivity for shuffle {shuffle_i+1} of {n_shuffles}")
     bad_surrogate_con.append(
         spectral_connectivity_epochs(
-            surrogate, method="imcoh", indices=indices, n_jobs=n_jobs, verbose=False
+            surrogate, method="imcoh", indices=indices, verbose=False
         )
     )