diff --git a/torchvision/datasets/celeba.py b/torchvision/datasets/celeba.py index 7ff0595166b..c15120af5a5 100644 --- 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 index 969642553a1..97a47c07beb 100644 --- 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)