(self)
| 96 | return {"input_ids": self.input_ids[idx], "attention_mask": self.attn_mask[idx]} |
| 97 | |
| 98 | def __len__(self): |
| 99 | return len(self.input_ids) |
| 100 | |
| 101 | test_ds = TestDataset(dataset, tokenizer, seqlen) |
| 102 | loader = DataLoader(test_ds, batch_size=batch_size) |
nothing calls this directly
no outgoing calls
no test coverage detected