def _obtain_peak_position(self, waveforms): shape = waveforms.shape significant_samples = self._extract_significant_entries(waveforms) sig_sam = significant_samples.astype(np.float32) sum_data = np.zeros_like(sig_sam) n = self.neighbours.astype(np.uint16) get_sum_array(sig_sam, sum_data, *shape, n, n.shape[0], self.lwt) return sum_data.argmax(2).astype(np.int)