From 5abfbb48254c62390e9f50df17fed64d0aa06bf4 Mon Sep 17 00:00:00 2001 From: GiannakopoulosIlias Date: Wed, 19 Jun 2024 11:32:10 -0400 Subject: [PATCH] Re-adding fastmri/models/feature_varnet.py to ensure CI passes - reinstalled black 22.3.0 --- fastmri/models/feature_varnet.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/fastmri/models/feature_varnet.py b/fastmri/models/feature_varnet.py index f9334566..343841d3 100644 --- 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,