Check for channel_indices is None

SpikeInterface · Jan 21, 2025 · 7bb12f0 · 7bb12f0
1 parent dafed80
commit 7bb12f0
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