Merge pull request #3636 from cwindolf/avg_chans_bug2

Check for channel_indices is None

Author: alejoe91

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