def __call__()

in src/chug/image/transforms_torch.py [0:0]

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

    def __call__(self, img):
        img_width, img_height = F.get_image_size(img)
        if (
            (self.input_size[0] > self.input_size[1] and img_width > img_height) or
            (self.input_size[0] < self.input_size[1] and img_width < img_height)
        ):
            img = F.rotate(img, angle=-90, expand=True, interpolation=self.interpolation)
        return img