Fix lint errors

Author: ProGamerGov

torchvision/datasets/_optical_flow.py

@@ -28,7 +28,7 @@ class FlowDataset(ABC, VisionDataset):
     # and it's up to whatever consumes the dataset to decide what valid_flow_mask should be.
     _has_builtin_flow_mask = False
 
-    def __init__(self, root, transforms=None) -> None:
+    def __init__(self, root, transforms=None):
 
         super().__init__(root=root)
         self.transforms = transforms
@@ -70,7 +70,7 @@ def __getitem__(self, index):
         else:
             return img1, img2, flow
 
-    def __len__(self):
+    def __len__(self) -> int:
         return len(self._image_list)
 
     def __rmul__(self, v):

torchvision/models/feature_extraction.py

@@ -55,7 +55,7 @@ class NodePathTracer(LeafModuleAwareTracer):
           _{int} is added. The counter starts from 1.
     """
 
-    def __init__(self, *args, **kwargs) -> None:
+    def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         # Track the qualified name of the Node being traced
         self.current_module_qualname = ""

torchvision/models/optical_flow/raft.py

@@ -27,7 +27,7 @@
 class ResidualBlock(nn.Module):
     """Slightly modified Residual block with extra relu and biases."""
 
-    def __init__(self, in_channels, out_channels, *, norm_layer, stride=1) -> None:
+    def __init__(self, in_channels, out_channels, *, norm_layer, stride=1):
         super().__init__()
 
         # Note regarding bias=True:
@@ -71,7 +71,7 @@ def forward(self, x):
 class BottleneckBlock(nn.Module):
     """Slightly modified BottleNeck block (extra relu and biases)"""
 
-    def __init__(self, in_channels, out_channels, *, norm_layer, stride=1) -> None:
+    def __init__(self, in_channels, out_channels, *, norm_layer, stride=1):
         super().__init__()
 
         # See note in ResidualBlock for the reason behind bias=True

torchvision/transforms/_transforms_video.py

@@ -26,7 +26,7 @@
 
 
 class RandomCropVideo(RandomCrop):
-    def __init__(self, size) -> None:
+    def __init__(self, size):
         if isinstance(size, numbers.Number):
             self.size = (int(size), int(size))
         else:
@@ -82,7 +82,7 @@ def __repr__(self) -> str:
 
 
 class CenterCropVideo:
-    def __init__(self, crop_size) -> None:
+    def __init__(self, crop_size):
         if isinstance(crop_size, numbers.Number):
             self.crop_size = (int(crop_size), int(crop_size))
         else: