Nifti saving function not working

[alinafdima]

deepali/src/deepali/utils/imageio/nifti.py

Line 116 in ae3e476

affine = grid.affine().cpu().numpy()

The Nifti saving function using nibabel here does not work because the affine matrix is incomplete. The output affine matrix from grid.affine() is 3x3 which is missing the origin/translation required for nibabel.