seriously

Author: NicolasHug

torchvision/datasets/celeba.py

@@ -105,7 +105,7 @@ def __init__(
         if mask == slice(None):  # if split == "all"
             self.filename = splits.index
         else:
-            self.filename = [splits.index[i] for i in torch.squeeze(torch.nonzero(mask))]
+            self.filename = [splits.index[i] for i in torch.squeeze(torch.nonzero(mask))]  # type: ignore[arg-type]
         self.identity = identity.data[mask]
         self.bbox = bbox.data[mask]
         self.landmarks_align = landmarks_align.data[mask]

torchvision/datasets/cityscapes.py

@@ -192,7 +192,7 @@ def __getitem__(self, index: int) -> Tuple[Any, Any]:
 
             targets.append(target)
 
-        target = tuple(targets) if len(targets) > 1 else targets[0]
+        target = tuple(targets) if len(targets) > 1 else targets[0]  # type: ignore[assignment]
 
         if self.transforms is not None:
             image, target = self.transforms(image, target)