linter

Author: NicolasHug

packaging/wheel/relocate.py

@@ -316,7 +316,9 @@ def patch_linux():
     output_dir = osp.join(PACKAGE_ROOT, "dist", ".wheel-process")
 
     image_binary = "image.so"
-    torchvision_binaries = [image_binary, ]
+    torchvision_binaries = [
+        image_binary,
+    ]
     for wheel in wheels:
         if osp.exists(output_dir):
             shutil.rmtree(output_dir)
@@ -351,7 +353,9 @@ def patch_win():
     output_dir = osp.join(PACKAGE_ROOT, "dist", ".wheel-process")
 
     image_binary = "image.pyd"
-    torchvision_binaries = [image_binary, ]
+    torchvision_binaries = [
+        image_binary,
+    ]
     for wheel in wheels:
         if osp.exists(output_dir):
             shutil.rmtree(output_dir)

torchvision/datasets/video_utils.py

@@ -5,7 +5,7 @@
 from typing import Any, Callable, cast, Optional, TypeVar, Union
 
 import torch
-from torchvision.io import  read_video, read_video_timestamps
+from torchvision.io import read_video, read_video_timestamps
 
 from .utils import tqdm
 

torchvision/io/__init__.py

@@ -15,7 +15,7 @@
     write_jpeg,
     write_png,
 )
-from .video import write_video, read_video, read_video_timestamps
+from .video import read_video, read_video_timestamps, write_video
 
 
 __all__ = [

torchvision/io/video.py

@@ -44,9 +44,11 @@ def _check_av_available() -> None:
     if isinstance(av, Exception):
         raise av
 
+
 def _av_available() -> bool:
     return not isinstance(av, Exception)
 
+
 # PyAV has some reference cycles
 _CALLED_TIMES = 0
 _GC_COLLECTION_INTERVAL = 10
@@ -172,6 +174,7 @@ def write_video(
         for packet in stream.encode():
             container.mux(packet)
 
+
 def _read_from_stream(
     container: "av.container.Container",
     start_offset: float,
@@ -292,7 +295,6 @@ def read_video(
     if output_format not in ("THWC", "TCHW"):
         raise ValueError(f"output_format should be either 'THWC' or 'TCHW', got {output_format}.")
 
-
     if True:  # ignore, this is to avoid a bigger diff in https://github.com/pytorch/vision/pull/9189
         _check_av_available()
 
@@ -437,4 +439,4 @@ def read_video_timestamps(filename: str, pts_unit: str = "pts") -> tuple[list[in
     if pts_unit == "sec":
         pts = [x * video_time_base for x in pts]
 
-    return pts, video_fps
+    return pts, video_fps