def __getitem__()

in torchnet/dataset/tensordataset.py [0:0]

• 8 lines of code
• 7 McCabe index (conditional complexity)

    def __getitem__(self, idx):
        super(TensorDataset, self).__getitem__(idx)
        if isinstance(self.data, dict):
            return {k: v[idx] for k, v in self.data.items()}
        elif isinstance(self.data, list):
            return [v[idx] for v in self.data]
        elif torch.is_tensor(self.data) or isinstance(self.data, np.ndarray):
            return self.data[idx]