Unverified Commit 3eb4333e authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

fix mypy on `VisionDataset` (#8134)

parent 83090c2a
......@@ -31,7 +31,7 @@ class FakeData(VisionDataset):
target_transform: Optional[Callable] = None,
random_offset: int = 0,
) -> None:
super().__init__(None, transform=transform, target_transform=target_transform) # type: ignore[arg-type]
super().__init__(transform=transform, target_transform=target_transform)
self.size = size
self.num_classes = num_classes
self.image_size = image_size
......
......@@ -29,7 +29,7 @@ class VisionDataset(data.Dataset):
def __init__(
self,
root: Optional[str] = None,
root: str = None, # type: ignore[assignment]
transforms: Optional[Callable] = None,
transform: Optional[Callable] = None,
target_transform: Optional[Callable] = None,
......
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