Ignore some mypy issues (#8516)

pytorch · Jul 8, 2024 · e24b3c4 · e24b3c4
1 parent e2aff7e
commit e24b3c4
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/torchvision/datasets/_optical_flow.py b/torchvision/datasets/_optical_flow.py
@@ -43,7 +43,7 @@ def __init__(self, root: Union[str, Path], transforms: Optional[Callable] = None
     def _read_img(self, file_name: str) -> Image.Image:
         img = Image.open(file_name)
         if img.mode != "RGB":
-            img = img.convert("RGB")
+            img = img.convert("RGB")  # type: ignore[assignment]
         return img
 
     @abstractmethod

diff --git a/torchvision/datasets/_stereo_matching.py b/torchvision/datasets/_stereo_matching.py
@@ -58,7 +58,7 @@ def __init__(self, root: Union[str, Path], transforms: Optional[Callable] = None
     def _read_img(self, file_path: Union[str, Path]) -> Image.Image:
         img = Image.open(file_path)
         if img.mode != "RGB":
-            img = img.convert("RGB")
+            img = img.convert("RGB")  # type: ignore [assignment]
         return img
 
     def _scan_pairs(

diff --git a/torchvision/datasets/widerface.py b/torchvision/datasets/widerface.py
@@ -92,7 +92,7 @@ def __getitem__(self, index: int) -> Tuple[Any, Any]:
         """
 
         # stay consistent with other datasets and return a PIL Image
-        img = Image.open(self.img_info[index]["img_path"])
+        img = Image.open(self.img_info[index]["img_path"])  # type: ignore[arg-type]
 
         if self.transform is not None:
             img = self.transform(img)

diff --git a/torchvision/utils.py b/torchvision/utils.py
@@ -612,8 +612,8 @@ def _parse_colors(
 
     colors = [ImageColor.getrgb(color) if isinstance(color, str) else color for color in colors]
     if dtype.is_floating_point:  # [0, 255] -> [0, 1]
-        colors = [tuple(v / 255 for v in color) for color in colors]
-    return colors
+        colors = [tuple(v / 255 for v in color) for color in colors]  # type: ignore[union-attr]
+    return colors  # type: ignore[return-value]
 
 
 def _log_api_usage_once(obj: Any) -> None: