diff --git a/mypy.ini b/mypy.ini index 0c46a09610b..dbcaab8770c 100644 --- a/mypy.ini +++ b/mypy.ini @@ -25,10 +25,6 @@ ignore_errors = True ignore_errors = True -[mypy-torchvision.models.detection.image_list] - -ignore_errors = True - [mypy-torchvision.models.detection.transform] ignore_errors = True diff --git a/torchvision/models/detection/image_list.py b/torchvision/models/detection/image_list.py index 333d3b569f2..58f32662b1a 100644 --- a/torchvision/models/detection/image_list.py +++ b/torchvision/models/detection/image_list.py @@ -10,14 +10,13 @@ class ImageList(object): varying sizes) as a single tensor. This works by padding the images to the same size, and storing in a field the original sizes of each image + + Args: + tensors (tensor): Tensor containing images. + image_sizes (list[tuple[int, int]]): List of Tuples each containing size of images. """ - def __init__(self, tensors: Tensor, image_sizes: List[Tuple[int, int]]): - """ - Args: - tensors (tensor) - image_sizes (list[tuple[int, int]]) - """ + def __init__(self, tensors: Tensor, image_sizes: List[Tuple[int, int]]) -> None: self.tensors = tensors self.image_sizes = image_sizes