Fix silence period and job_kwargs

src/spikeinterface/preprocessing/silence_periods.py

@@ -5,8 +5,9 @@
 from spikeinterface.core.core_tools import define_function_handling_dict_from_class
 from .basepreprocessor import BasePreprocessor, BasePreprocessorSegment
 
-from spikeinterface.core import get_random_data_chunks, get_noise_levels
+from spikeinterface.core import get_noise_levels
 from spikeinterface.core.generate import NoiseGeneratorRecording
+from spikeinterface.core.job_tools import split_job_kwargs
 
 
 class SilencedPeriodsRecording(BasePreprocessor):
@@ -36,6 +37,12 @@ class SilencedPeriodsRecording(BasePreprocessor):
         - "noise": The periods are filled with a gaussion noise that has the
                    same variance that the one in the recordings, on a per channel
                    basis
+    job_kwargs : dict
+        Keyword arguments for the joblib parallelization. If you want to use
+        `job_kwargs`, you need to pass them as a dictionary with the key "job_kwargs".
+        For example, `job_kwargs={"num_workers": 4}`.
+        Note that this is not used for the `get_noise_levels` function, which has its own
+        `random_slices_kwargs` argument.
     **random_chunk_kwargs : Keyword arguments for `spikeinterface.core.get_random_data_chunk()` function
 
     Returns
@@ -44,7 +51,16 @@ class SilencedPeriodsRecording(BasePreprocessor):
         The recording extractor after silencing some periods
     """
 
-    def __init__(self, recording, list_periods, mode="zeros", noise_levels=None, seed=None, **random_chunk_kwargs):
+    def __init__(
+        self,
+        recording,
+        list_periods,
+        mode="zeros",
+        noise_levels=None,
+        seed=None,
+        job_kwargs=dict(),
+        **random_chunk_kwargs,
+    ):
         available_modes = ("zeros", "noise")
         num_seg = recording.get_num_segments()
 
@@ -74,7 +90,7 @@ def __init__(self, recording, list_periods, mode="zeros", noise_levels=None, see
                 random_slices_kwargs = random_chunk_kwargs.copy()
                 random_slices_kwargs["seed"] = seed
                 noise_levels = get_noise_levels(
-                    recording, return_scaled=False, random_slices_kwargs=random_slices_kwargs
+                    recording, return_scaled=False, random_slices_kwargs=random_slices_kwargs, **job_kwargs
                 )
             noise_generator = NoiseGeneratorRecording(
                 num_channels=recording.get_num_channels(),
@@ -97,7 +113,9 @@ def __init__(self, recording, list_periods, mode="zeros", noise_levels=None, see
             rec_segment = SilencedPeriodsRecordingSegment(parent_segment, periods, mode, noise_generator, seg_index)
             self.add_recording_segment(rec_segment)
 
-        self._kwargs = dict(recording=recording, list_periods=list_periods, mode=mode, seed=seed)
+        self._kwargs = dict(
+            recording=recording, list_periods=list_periods, mode=mode, seed=seed, noise_levels=noise_levels
+        )
         self._kwargs.update(random_chunk_kwargs)