seriously

pytorch · Oct 15, 2024 · 6a7b566 · 6a7b566
1 parent d8d9c85
commit 6a7b566
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/torchvision/datasets/celeba.py b/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]

diff --git a/torchvision/datasets/cityscapes.py b/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)