Skip to content

Commit

Permalink
fixes tests
Browse files Browse the repository at this point in the history
Signed-off-by: Wenqi Li <[email protected]>
  • Loading branch information
wyli committed Jan 17, 2023
1 parent 10e420e commit 690d075
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions monai/transforms/spatial/array.py
Original file line number Diff line number Diff line change
Expand Up @@ -1224,7 +1224,7 @@ def inverse_transform(self, data: torch.Tensor, transform) -> torch.Tensor:
out: torch.Tensor = xform(img_t.unsqueeze(0), transform_t, spatial_size=sp_size).float().squeeze(0)
out = convert_to_dst_type(out, dst=data, dtype=out.dtype)[0]
if isinstance(data, MetaTensor):
self.update_meta(out, transform_t)
out.affine @= self.update_meta(out, transform_t) # type: ignore
return out


Expand Down Expand Up @@ -2756,7 +2756,7 @@ def __call__(
"do_resampling": do_resampling,
},
)
out.affine = self.update_meta(out, mat, img.shape[1:], sp_size) # type: ignore
out.affine @= self.update_meta(out, mat, img.shape[1:], sp_size) # type: ignore
return out

def lazy_call(self, img, affine, output_size, mode, padding_mode, do_resampling) -> torch.Tensor:
Expand Down Expand Up @@ -2798,7 +2798,7 @@ def inverse(self, data: torch.Tensor) -> torch.Tensor:
if not isinstance(out, MetaTensor):
out = MetaTensor(out)
out.meta = data.meta # type: ignore
self.update_meta(out, inv_affine, data.shape[1:], orig_size)
out.affine @= self.update_meta(out, inv_affine, data.shape[1:], orig_size)
return out


Expand Down

0 comments on commit 690d075

Please sign in to comment.