Merge pull request #3636 from cwindolf/avg_chans_bug2

Check for channel_indices is None
SpikeInterface · Jan 21, 2025 · 749a349 · 749a349
2 parents d0919ad + 7bb12f0
commit 749a349
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/spikeinterface/preprocessing/average_across_direction.py b/src/spikeinterface/preprocessing/average_across_direction.py
@@ -132,7 +132,10 @@ def get_traces(self, start_frame, end_frame, channel_indices):
         # now, divide by the number of channels at that position
         traces /= self.n_chans_each_pos
 
-        return traces[:, channel_indices]
+        if channel_indices is not None:
+            traces = traces[:, channel_indices]
+
+        return traces
 
 
 # function for API