format signatures with black (#2070)

cd77d17a · Philip Meier · GitHub · 9a3afff4 · cd77d17a
Unverified Commit cd77d17a authored Apr 07, 2020 by Philip Meier Committed by GitHub Apr 07, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 20 additions and 7 deletions

torchvision/utils.py torchvision/utils.py +20 -7

No files found.
--- a/torchvision/utils.py
+++ b/torchvision/utils.py
@@ -6,10 +6,15 @@ import math
 irange = range
-def make_grid(tensor: Union[torch.Tensor, Sequence[torch.Tensor]], nrow: int = 8,
+def make_grid(
-              padding: int = 2, normalize: bool = False,
+    tensor: Union[torch.Tensor, Sequence[torch.Tensor]],
-              range: Optional[Tuple[int, int]] = None, scale_each: bool = False,
+    nrow: int = 8,
-              pad_value: int = 0) -> torch.Tensor:
+    padding: int = 2,
+    normalize: bool = False,
+    range: Optional[Tuple[int, int]] = None,
+    scale_each: bool = False,
+    pad_value: int = 0,
+) -> torch.Tensor:
    """Make a grid of images.
    Args:
@@ -93,9 +98,17 @@ def make_grid(tensor: Union[torch.Tensor, Sequence[torch.Tensor]], nrow: int = 8
    return grid
-def save_image(tensor: Union[torch.Tensor, Sequence[torch.Tensor]], fp: Union[Text, pathlib.Path, BinaryIO],
+def save_image(
-               nrow: int = 8, padding: int = 2, normalize: bool = False, range: Optional[Tuple[int, int]] = None,
+    tensor: Union[torch.Tensor, Sequence[torch.Tensor]],
-               scale_each: bool = False, pad_value: int = 0, format: Optional[str] = None) -> None:
+    fp: Union[Text, pathlib.Path, BinaryIO],
+    nrow: int = 8,
+    padding: int = 2,
+    normalize: bool = False,
+    range: Optional[Tuple[int, int]] = None,
+    scale_each: bool = False,
+    pad_value: int = 0,
+    format: Optional[str] = None,
+) -> None:
    """Save a given Tensor into an image file.
    Args: