Re-adding fastmri/models/feature_varnet.py to ensure CI passes - rein…

…stalled black 22.3.0
facebookresearch · Jul 23, 2024 · 280a7a8 · 280a7a8
1 parent cb0142a
commit 280a7a8
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/fastmri/models/feature_varnet.py b/fastmri/models/feature_varnet.py
@@ -53,9 +53,9 @@ def image_uncrop(image: Tensor, original_image: Tensor) -> Tensor:
     pad_height_left, pad_width = _calc_uncrop(image.shape[-1], in_shape[-1])
 
     try:
-        original_image[..., pad_height_top:pad_height, pad_height_left:pad_width] = (
-            image[...]
-        )
+        original_image[
+            ..., pad_height_top:pad_height, pad_height_left:pad_width
+        ] = image[...]
     except RuntimeError:
         print(f"in_shape: {in_shape}, image shape: {image.shape}")
         raise
@@ -553,7 +553,6 @@ def forward(self, image: torch.Tensor) -> torch.Tensor:
 
 
 class NormUnet(nn.Module):
-
     def __init__(
         self,
         chans: int,
@@ -644,7 +643,6 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
 
 
 class Norm1DUnet(nn.Module):
-
     def __init__(
         self,
         chans: int,