Error when preprocessing with n_jets: -1

Using n_jets: -1 in the preprocessing config yields the below error.

Traceback (most recent call last):
  File "/unix/atlastracking/svanstroud/miniconda3/envs/umami/bin/preprocessing.py", line 180, in <module>
    sampler.Run()
  File "/unix/atlastracking/svanstroud/umami/umami/preprocessing_tools/resampling/count_sampling.py", line 187, in Run
    self.get_indices()
  File "/unix/atlastracking/svanstroud/umami/umami/preprocessing_tools/resampling/count_sampling.py", line 132, in get_indices
    indices_to_keep[class_category] = rng.choice(
  File "_generator.pyx", line 775, in numpy.random._generator.Generator.choice
ValueError: negative dimensions are not allowed