runtime error

/s] Generating validation split: 25%|██▌ | 1000/3925 [00:00<00:01, 1837.47 examples/s] Generating validation split: 35%|███▌ | 1375/3925 [00:00<00:01, 2283.84 examples/s] Generating validation split: 44%|████▍ | 1735/3925 [00:00<00:00, 2612.63 examples/s] Generating validation split: 55%|█████▌ | 2161/3925 [00:01<00:01, 1551.18 examples/s] Generating validation split: 66%|██████▌ | 2573/3925 [00:01<00:00, 1963.22 examples/s] Generating validation split: 76%|███████▋ | 3000/3925 [00:01<00:00, 1579.37 examples/s] Generating validation split: 86%|████████▋ | 3392/3925 [00:01<00:00, 1928.74 examples/s] Generating validation split: 97%|█████████▋| 3803/3925 [00:01<00:00, 2312.33 examples/s] Generating validation split: 100%|██████████| 3925/3925 [00:04<00:00, 964.61 examples/s] Traceback (most recent call last): File "/home/user/app/app.py", line 74, in <module> embedding_img() File "/home/user/app/app.py", line 36, in embedding_img images = [imagenette[i]['image'] for i in sample_idx] File "/home/user/app/app.py", line 36, in <listcomp> images = [imagenette[i]['image'] for i in sample_idx] File "/home/user/.local/lib/python3.10/site-packages/datasets/arrow_dataset.py", line 2803, in __getitem__ return self._getitem(key) File "/home/user/.local/lib/python3.10/site-packages/datasets/arrow_dataset.py", line 2787, in _getitem pa_subtable = query_table(self._data, key, indices=self._indices if self._indices is not None else None) File "/home/user/.local/lib/python3.10/site-packages/datasets/formatting/formatting.py", line 583, in query_table _check_valid_index_key(key, size) File "/home/user/.local/lib/python3.10/site-packages/datasets/formatting/formatting.py", line 526, in _check_valid_index_key raise IndexError(f"Invalid key: {key} is out of bounds for size {size}") IndexError: Invalid key: 9469 is out of bounds for size 9469

Container logs: