def __getitem__()

in distilbertqatrain.py [0:0]

• 2 lines of code
• 2 McCabe index (conditional complexity)

    def __getitem__(self, idx):
        return {key: torch.tensor(val[idx]) for key, val in self.encodings.items()}