可以调整一下上面代码的对齐位置吗

非常抱歉，好像我的代码对齐有些问题。以下是调整过后的代码：

import gzip

def load_nii_data(file_path):
if file_path.endswith('.gz'):
with gzip.open(file_path, 'rb') as f:
image = sitk.ReadImage(f.read())
else:
image = sitk.ReadImage(file_path)
image_data = sitk.GetArrayFromImage(image)
return image_data

希望这样更易读一些，如果还有问题，欢迎指出，我会及时修改。