Unverified Commit 835c2c35 authored by F-G Fernandez's avatar F-G Fernandez Committed by GitHub
Browse files

Added missing typing annotations in datasets/coco (#4168)

parent d993ce56
......@@ -26,7 +26,7 @@ class CocoDetection(VisionDataset):
transform: Optional[Callable] = None,
target_transform: Optional[Callable] = None,
transforms: Optional[Callable] = None,
):
) -> None:
super().__init__(root, transforms, transform, target_transform)
from pycocotools.coco import COCO
......@@ -37,7 +37,7 @@ class CocoDetection(VisionDataset):
path = self.coco.loadImgs(id)[0]["file_name"]
return Image.open(os.path.join(self.root, path)).convert("RGB")
def _load_target(self, id) -> List[Any]:
def _load_target(self, id: int) -> List[Any]:
return self.coco.loadAnns(self.coco.getAnnIds(id))
def __getitem__(self, index: int) -> Tuple[Any, Any]:
......@@ -95,5 +95,5 @@ class CocoCaptions(CocoDetection):
"""
def _load_target(self, id) -> List[str]:
def _load_target(self, id: int) -> List[str]:
return [ann["caption"] for ann in super()._load_target(id)]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment