From ffcdf5a8f1eadc7cd7d8018b553163f375de7632 Mon Sep 17 00:00:00 2001 From: Baptiste Grimaud <83828302+b-grimaud@users.noreply.github.com> Date: Mon, 18 Nov 2024 18:08:23 +0100 Subject: [PATCH] Expose unit_locations as property --- src/spikeinterface/extractors/herdingspikesextractors.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/spikeinterface/extractors/herdingspikesextractors.py b/src/spikeinterface/extractors/herdingspikesextractors.py index 2deb9fc2a8..a83fbbb838 100644 --- 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")