Expose unit_locations as property

SpikeInterface · Nov 18, 2024 · ffcdf5a · ffcdf5a
1 parent b0ff928
commit ffcdf5a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/spikeinterface/extractors/herdingspikesextractors.py b/src/spikeinterface/extractors/herdingspikesextractors.py
@@ -45,11 +45,13 @@ def __init__(self, file_path, load_unit_info=True):
         spike_times = self._rf["times"][()]
         unit_locs = self._rf["centres"][()]
 
+        self.unit_locations = unit_locs
+
         # if load_unit_info:
         #     self.load_unit_info()
 
         BaseSorting.__init__(self, sampling_frequency, unit_ids)
-        self.add_sorting_segment(HerdingspikesSortingSegment(unit_ids, spike_times, spike_ids, unit_locs))
+        self.add_sorting_segment(HerdingspikesSortingSegment(unit_ids, spike_times, spike_ids, self.unit_locations))
         self._kwargs = {"file_path": str(Path(file_path).absolute()), "load_unit_info": load_unit_info}
 
         self.extra_requirements.append("h5py")