'datasets.arrow_dataset.Dataset' do not support numpy.int64 as index?

[qianyue76]

When I predict the model
`from setfit import SetFitModel
from datasets import load_dataset
model = SetFitModel.from_pretrained(model_name)

test_dataset = load_dataset("csv", data_files="dataset.csv",split="train")
predictions = model.predict(test_dataset)`

I meet the error

I print the type of length_sorted_idx and sentences in the top 2 lines.
And my relevant package version as follows:
setfit 1.1.0.dev0
sentence-transformers 3.1.0
datasets 3.0.0
torch 2.0.1
My question is on the title.

[antrec]

+1 (Should this not be an issue on the datasets package) ?

EDIT: Has it not been fixed here huggingface/datasets@0ec973d ? Yet you are using datasets 3.0.0 which includes this PR 🤔 ...