def resize_input()

in src/image_gen_aux/preprocessors/teed/teed.py [0:0]

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

    def resize_input(self, tensor):
        t_shape = tensor.shape
        if t_shape[2] % 8 != 0 or t_shape[3] % 8 != 0:
            img_w = ((t_shape[3] // 8) + 1) * 8
            img_h = ((t_shape[2] // 8) + 1) * 8
            new_tensor = F.interpolate(tensor, size=(img_h, img_w), mode="bicubic", align_corners=False)
        else:
            new_tensor = tensor
        return new_tensor