from __future__ import annotations from torch.utils.data import Dataset from sentence_transformers import SentenceTransformer from sentence_transformers.readers.InputExample import InputExample class SentencesDataset(Dataset): """ DEPRECATED: This class is no longer used. Instead of wrapping your List of InputExamples in a SentencesDataset and then passing it to the DataLoader, you can pass the list of InputExamples directly to the dataset loader. """ def __init__(self, examples: list[InputExample], model: SentenceTransformer): self.examples = examples def __getitem__(self, item): return self.examples[item] def __len__(self): return len(self.examples)