runtime error

Exit code: 1. Reason: Generating train split: 0%| | 0/94 [00:00<?, ? examples/s] Generating train split: 100%|██████████| 94/94 [00:00<00:00, 3264.62 examples/s] Traceback (most recent call last): File "/home/user/app/app.py", line 86, in <module> main() File "/home/user/app/app.py", line 76, in main train_dataset, val_dataset = load_and_preprocess_data() File "/home/user/app/app.py", line 10, in load_and_preprocess_data train_dataset, val_dataset = train_test_split(dataset['train'], test_size=0.2, random_state=42) File "/usr/local/lib/python3.10/site-packages/sklearn/utils/_param_validation.py", line 216, in wrapper return func(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/sklearn/model_selection/_split.py", line 2876, in train_test_split return list( File "/usr/local/lib/python3.10/site-packages/sklearn/model_selection/_split.py", line 2878, in <genexpr> (_safe_indexing(a, train), _safe_indexing(a, test)) for a in arrays File "/usr/local/lib/python3.10/site-packages/sklearn/utils/_indexing.py", line 270, in _safe_indexing return _array_indexing(X, indices, indices_dtype, axis=axis) File "/usr/local/lib/python3.10/site-packages/sklearn/utils/_indexing.py", line 36, in _array_indexing return array[key, ...] if axis == 0 else array[:, key] File "/usr/local/lib/python3.10/site-packages/datasets/arrow_dataset.py", line 2780, in __getitem__ return self._getitem(key) File "/usr/local/lib/python3.10/site-packages/datasets/arrow_dataset.py", line 2764, in _getitem pa_subtable = query_table(self._data, key, indices=self._indices) File "/usr/local/lib/python3.10/site-packages/datasets/formatting/formatting.py", line 593, in query_table _check_valid_index_key(key, size) File "/usr/local/lib/python3.10/site-packages/datasets/formatting/formatting.py", line 543, in _check_valid_index_key _check_valid_index_key(int(max(key)), size=size) TypeError: '<' not supported between instances of 'int' and 'ellipsis'

Container logs: