def __getitem__()

in ASLRecognition/scripts/train.py [0:0]

• 6 lines of code
• 1 McCabe index (conditional complexity)

    def __getitem__(self, i):
        image = Image.open(self.X[i])
        image = self.aug(image)
        image = np.transpose(image, (2, 0, 1)).astype(np.float32)
        label = self.y[i]
        return torch.tensor(image, dtype=torch.float), torch.tensor(label, dtype=torch.long)