Merge pull request #196 from ttngu207/datajoint-spikeinterface

ttngu207 · web-flow · commit a22561a08fb2 · 2024-07-05T17:01:57.000-05:00
fix: `spikes` object no longer available from `ComputeSpikeLocations`…
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):`