def _setup_size()

in utils/mytransforms.py [0:0]

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

def _setup_size(size, error_msg): #from torchvision.transforms.transforms (for Pytorch compatibility)
    if isinstance(size, numbers.Number):
        return int(size), int(size)

    if isinstance(size, Sequence) and len(size) == 1:
        return size[0], size[0]

    if len(size) != 2:
        raise ValueError(error_msg)

    return size