Saving noise levels

SpikeInterface · Dec 27, 2023 · 1863308 · 1863308
1 parent 9f24c15
commit 1863308
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/spikeinterface/preprocessing/silence_periods.py b/src/spikeinterface/preprocessing/silence_periods.py
@@ -40,7 +40,6 @@ class SilencedPeriodsRecording(BasePreprocessor):
     name = "silence_periods"
 
     def __init__(self, recording, list_periods, mode="zeros", **random_chunk_kwargs):
-        import scipy.interpolate
 
         available_modes = ("zeros", "noise")
         num_seg = recording.get_num_segments()
@@ -71,6 +70,7 @@ def __init__(self, recording, list_periods, mode="zeros", **random_chunk_kwargs)
         else:
             noise_levels = None
 
+        self.noise_levels = noise_levels
         BasePreprocessor.__init__(self, recording)
         for seg_index, parent_segment in enumerate(recording._recording_segments):
             periods = list_periods[seg_index]
@@ -80,7 +80,7 @@ def __init__(self, recording, list_periods, mode="zeros", **random_chunk_kwargs)
             self.add_recording_segment(rec_segment)
 
         self._kwargs = dict(
-            recording=recording.to_dict(), list_periods=list_periods, mode=mode, noise_levels=noise_levels
+            recording=recording, list_periods=list_periods, mode=mode, noise_levels=noise_levels
         )