Skip to content

Commit 284edca

Browse files
authored
Fix attribute not exist error
1 parent bc3bb2b commit 284edca

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

kilosort/io.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -651,7 +651,7 @@ def __init__(self, filename, n_chan_bin: int, fs: int = 30000,
651651
a, b = self._get_batch_edges(self.n_batches_raw-1)
652652
batch_size = b - a - self.nt
653653
if batch_size < self.nt:
654-
self.n_batches -= 1
654+
self.n_batches_raw -= 1
655655
self.imax -= batch_size
656656

657657
self.set_downsampling(batch_downsampling)

0 commit comments

Comments
 (0)