Hi,
set format of the embeddings column to NumPy after the map call to get a NumPy array:
embeddings_dataset.set_format("numpy", columns=["embeddings"], output_all_columns=True)
type(embeddings_test[0]["embeddings"])