Skip to content

Commit 03e8cbb

Browse files
authored
fix: Too strict assertion in Grid.resize() (#165)
1 parent bdc0852 commit 03e8cbb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/deepali/core/grid.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1095,11 +1095,11 @@ def _resize(self, size: Tensor, align_corners: Optional[bool] = None) -> Grid:
10951095
if align_corners:
10961096
spacing = (self.extent() - self.spacing()) / (size - 1)
10971097
grid._spacing = torch.where(self._size.gt(0), spacing, self._spacing)
1098-
assert torch.allclose(grid.origin(), self.origin())
1098+
assert torch.allclose(grid.origin(), self.origin(), rtol=1e-4, atol=1e-6)
10991099
else:
11001100
spacing = self.extent() / size
11011101
grid._spacing = torch.where(self._size.gt(0), spacing, self._spacing)
1102-
assert torch.allclose(grid.extent(), self.extent())
1102+
assert torch.allclose(grid.extent(), self.extent(), rtol=1e-4, atol=1e-6)
11031103
return grid
11041104

11051105
def resize(

0 commit comments

Comments
 (0)