fix: spikes object no longer available from ComputeSpikeLocations (SpikeInterface/spikeinterface#3015)

ttngu207 · ttngu207 · commit 1f05998e25d8 · 2024-07-05T16:45:40.000-05:00
diff --git a/element_array_ephys/ephys_no_curation.py b/element_array_ephys/ephys_no_curation.py
@@ -1073,7 +1073,9 @@ def make(self, key):
             }
 
             spike_locations = sorting_analyzer.get_extension("spike_locations")
-            spikes_df = pd.DataFrame(spike_locations.spikes)
+            extremum_channel_inds = si.template_tools.get_template_extremum_channel(sorting_analyzer, outputs="index")
+            spikes_df = pd.DataFrame(
+                sorting_analyzer.sorting.to_spike_vector(extremum_channel_inds=extremum_channel_inds))
 
             units = []
             for unit_idx, unit_id in enumerate(si_sorting.unit_ids):

Original file line number	Diff line number	Diff line change
`@@ -1073,7 +1073,9 @@ def make(self, key):`
`1073`	`1073`	`}`
`1074`	`1074`
`1075`	`1075`	`spike_locations = sorting_analyzer.get_extension("spike_locations")`
`1076`		`- spikes_df = pd.DataFrame(spike_locations.spikes)`
	`1076`	`+ extremum_channel_inds = si.template_tools.get_template_extremum_channel(sorting_analyzer, outputs="index")`
	`1077`	`+ spikes_df = pd.DataFrame(`
	`1078`	`+ sorting_analyzer.sorting.to_spike_vector(extremum_channel_inds=extremum_channel_inds))`
`1077`	`1079`
`1078`	`1080`	`units = []`
`1079`	`1081`	`for unit_idx, unit_id in enumerate(si_sorting.unit_ids):`