datasets 4.0.0 compatibility

[mengerj]

I recently got an error when running the SentenceTransformer.encode() method, when passing a column from a DatasetDict:
TypeError: Wrong key type: '0' of type '<class 'numpy.int64'>'. Expected one of int, slice, range, str or Iterable.
I found that this only occurs with datasets 4.0.0, while newer and older versions work. Just letting other know..

[tomaarsen]

Hello!

There were indeed a few compatibility issues with datasets v4.0, but fixes were introduced in Sentence Transformers v5.1.0 (https://github.com/huggingface/sentence-transformers/releases/tag/v5.1.0). Consider upgrading Sentence Transformers with pip install -U sentence-transformers and your issues should resolve themselves.
If not, please let me know!

• Tom Aarsen