def __len__()

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

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

    def __len__(self):
        if isinstance(self.data, dict):
            return len(list(self.data.values())[0])
        elif isinstance(self.data, list):
            return len(self.data[0])
        elif torch.is_tensor(self.data) or isinstance(self.data, np.ndarray):
            return len(self.data)