IndexError issue

[alejoe91]

Hi @cwindolf

Got this error one two datasets:

Error log:
  index 4733012 is out of bounds for axis 0 with size 4733012

Command error:
          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
      )
      ^
    File "/opt/venv/lib/python3.13/site-packages/dartsort/peel/peel_base.py", line 352, in peel
      for result, chunk_start_samples in zip(
                                         ~~~^
          results, chunks_to_do, strict=True
          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
      ):
      ^
    File "/opt/venv/lib/python3.13/site-packages/dartsort/util/logging_util.py", line 180, in __iter__
      for o in it:
               ^^
    File "/opt/venv/lib/python3.13/site-packages/dartsort/util/multiprocessing_util.py", line 78, in map
      for result in map(function, *iterables):
                    ~~~^^^^^^^^^^^^^^^^^^^^^^
    File "/opt/venv/lib/python3.13/site-packages/dartsort/peel/peel_base.py", line 1279, in _peeler_process_job
      return _peeler_process_context.ctx.peeler.process_chunk(
             ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^
          chunk_start_samples,
          ^^^^^^^^^^^^^^^^^^^^
      ...<4 lines>...
          to_cpu=_peeler_process_context.ctx.to_cpu,
          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
      )
      ^
    File "/opt/venv/lib/python3.13/site-packages/dartsort/peel/peel_base.py", line 520, in process_chunk
      chunk_result = self.featurize_chunk_result(
          peel_result=peel_result,
      ...<4 lines>...
          n_resid_snips=n_resid_snips,
      )
    File "/opt/venv/lib/python3.13/site-packages/dartsort/peel/peel_base.py", line 573, in featurize_chunk_result
      chunk_end_s = self.recording.sample_index_to_time(chunk_end_samples)
    File "/opt/venv/lib/python3.13/site-packages/spikeinterface/core/time_series.py", line 307, in sample_index_to_time
      return rs.sample_index_to_time(sample_ind)
             ~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^
    File "/opt/venv/lib/python3.13/site-packages/spikeinterface/core/time_series.py", line 502, in sample_index_to_time
      time_s = self.time_vector[sample_ind]
               ~~~~~~~~~~~~~~~~^^^^^^^^^^^^
    File "/opt/venv/lib/python3.13/site-packages/numpy/_core/memmap.py", line 360, in __getitem__
      res = super().__getitem__(index)
  IndexError: index 4733012 is out of bounds for axis 0 with size 4733012

  During handling of the above exception, another exception occurred:

  Traceback (most recent call last):
    File "/tmp/nxf.HUZc7eIwp0/capsule/code/run_capsule.py", line 378, in <module>
      raise Exception(e)
  Exception: index 4733012 is out of bounds for axis 0 with size 4733012

Work dir:
  s3://codeocean-s3batchbucket-16itpvq060udk/24752a5a-c29d-4c19-86fd-3d265695c09d/f1/40b4ad14225e08d910d308061a098a

Tip: view the complete command output by changing to the process work dir and entering the command `cat .command.out`


WARN: Killing running tasks (34)

Can you take a look? Thanks!