def __init__()

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

• 9 lines of code
• 4 McCabe index (conditional complexity)

    def __init__(self, dataset, transforms):
        super(TransformDataset, self).__init__()

        assert isinstance(transforms, dict) or callable(transforms), \
            'expected a dict of transforms or a function'
        if isinstance(transforms, dict):
            for k, v in transforms.items():
                assert callable(v), str(k) + ' is not a function'

        self.dataset = dataset
        self.transforms = transforms