Merge pull request #3692 from alejoe91/add-missing-parents

Add `_parent` to `select_segment` classes

Author: samuelgarcia

src/spikeinterface/core/channelslice.py

@@ -25,11 +25,10 @@ def __init__(self, parent_recording, channel_ids=None, renamed_channel_ids=None)
                 np.unique(renamed_channel_ids)
             ), "renamed_channel_ids must be unique!"
 
-        self._parent_recording = parent_recording
         self._channel_ids = np.asarray(channel_ids)
         self._renamed_channel_ids = np.asarray(renamed_channel_ids)
 
-        parents_chan_ids = self._parent_recording.get_channel_ids()
+        parents_chan_ids = parent_recording.get_channel_ids()
 
         # some checks
         assert all(
@@ -54,7 +53,7 @@ def __init__(self, parent_recording, channel_ids=None, renamed_channel_ids=None)
         self._parent_channel_indices = parent_recording.ids_to_indices(self._channel_ids)
 
         # link recording segment
-        for parent_segment in self._parent_recording._recording_segments:
+        for parent_segment in parent_recording._recording_segments:
             sub_segment = ChannelSliceRecordingSegment(parent_segment, self._parent_channel_indices)
             self.add_recording_segment(sub_segment)
 

src/spikeinterface/core/segmentutils.py

@@ -243,6 +243,7 @@ def __init__(self, recording: BaseRecording, segment_indices: int | list[int]):
         for segment_index in segment_indices:
             rec_seg = recording._recording_segments[segment_index]
             self.add_recording_segment(rec_seg)
+        self._parent = recording
 
         self._kwargs = {"recording": recording, "segment_indices": segment_indices}
 
@@ -564,6 +565,7 @@ def __init__(self, parent_sorting: BaseSorting, recording_or_recording_list=None
             )
             sliced_segment = sliced_parent_sorting._sorting_segments[0]
             self.add_sorting_segment(sliced_segment)
+        self._parent = parent_sorting
 
         self._kwargs = {"parent_sorting": parent_sorting, "recording_or_recording_list": recording_list}
 

src/spikeinterface/preprocessing/basepreprocessor.py

@@ -9,7 +9,6 @@ class BasePreprocessor(BaseRecording):
     def __init__(self, recording, sampling_frequency=None, channel_ids=None, dtype=None):
         assert isinstance(recording, BaseRecording), "'recording' must be a RecordingExtractor"
 
-        self._parent_recording = recording
         if sampling_frequency is None:
             sampling_frequency = recording.get_sampling_frequency()
         if channel_ids is None: