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

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.