Apply comments from code review

shimmingtoolbox/unwrap/prelude.py

@@ -81,8 +81,9 @@ def prelude(wrapped_phase, mag, affine, mask=None, threshold=None, is_unwrapping
 
     fname_phase_unwrapped = glob.glob(os.path.join(path_tmp, 'rawPhase_unwrapped*'))[0]
 
+    # When loading fname_phase_unwrapped, if a singleton is on the last dimension, it will not appear in the last
+    # dimension in phase_unwrapped. To be consistent with the size of the input, the singleton are added back.
     phase_unwrapped = nib.load(fname_phase_unwrapped).get_fdata()
-
     for _ in range(wrapped_phase.ndim - phase_unwrapped.ndim):
         phase_unwrapped = np.expand_dims(phase_unwrapped, -1)