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