Merge branch 'main' into main

SpikeInterface · Oct 17, 2023 · 82ee456 · 82ee456
2 parents 44ad0ef + 4c76371
commit 82ee456
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/spikeinterface/sortingcomponents/clustering/split.py b/src/spikeinterface/sortingcomponents/clustering/split.py
@@ -59,7 +59,7 @@ def split_clusters(
 
     job_kwargs = fix_job_kwargs(job_kwargs)
     n_jobs = job_kwargs["n_jobs"]
-    mp_context = job_kwargs["mp_context"]
+    mp_context = job_kwargs.get("mp_context", None)
     progress_bar = job_kwargs["progress_bar"]
     max_threads_per_process = job_kwargs["max_threads_per_process"]
 
@@ -72,7 +72,7 @@ def split_clusters(
     with Executor(
         max_workers=n_jobs,
         initializer=split_worker_init,
-        mp_context=get_context(mp_context),
+        mp_context=get_context(method=mp_context),
         initargs=(recording, features_dict_or_folder, original_labels, method, method_kwargs, max_threads_per_process),
     ) as pool:
         labels_set = np.setdiff1d(peak_labels, [-1])