propgate to children

Author: h-mayorquin

src/spikeinterface/preprocessing/basepreprocessor.py

@@ -21,6 +21,13 @@ def __init__(self, recording, sampling_frequency=None, channel_ids=None, dtype=N
         recording.copy_metadata(self, only_main=False, ids=channel_ids)
         self._parent = recording
 
+        # Propagate the attached probegroup. copy_metadata only handles annotations
+        # and properties; `_probegroup` is a direct attribute and needs its own path.
+        # Subclasses that change channels (e.g. slicing) should override by slicing
+        # the probegroup themselves via set_probegroup.
+        if getattr(recording, "_probegroup", None) is not None and channel_ids is None:
+            self._probegroup = recording._probegroup
+
         # self._kwargs have to be handled in subclass
 
 

src/spikeinterface/preprocessing/tests/test_interpolate_bad_channels.py

@@ -126,7 +126,7 @@ def test_compare_input_argument_ranges_against_ibl(shanks, p, sigma_um, num_chan
     # distribute default probe locations across 4 shanks if set
     rng = np.random.default_rng(seed=None)
     x = rng.choice(shanks, num_channels)
-    probe = recording.get_probegroup().probes[0]
+    probe = recording._probegroup.probes[0]
     probe._contact_positions[:, 0] = x
     recording._probegroup._build_contact_vector()
     recording.set_property("location", recording.get_channel_locations())
@@ -172,7 +172,7 @@ def test_output_values():
         [5, 5, 5, 7, 3],
     ]  # all others equal distance away.
     # Overwrite the probe information with the new locations
-    probe = recording.get_probegroup().probes[0]
+    probe = recording._probegroup.probes[0]
     for idx, (x, y) in enumerate(zip(*new_probe_locs)):
         probe._contact_positions[idx, 0] = x
         probe._contact_positions[idx, 1] = y
@@ -191,7 +191,7 @@ def test_output_values():
     # Shift the last channel position so that it is 4 units, rather than 2
     # away. Setting sigma_um = p = 1 allows easy calculation of the expected
     # weights.
-    probe = recording.get_probegroup().probes[0]
+    probe = recording._probegroup.probes[0]
     probe._contact_positions[-1, 0] = 5
     probe._contact_positions[-1, 1] = 9
     recording._probegroup._build_contact_vector()