Fix for DWI None mask export

UKRIN-MAPS · Nov 28, 2023 · 3d06a8a · 3d06a8a
1 parent a0c3666
commit 3d06a8a
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/ukat/mapping/diffusion.py b/ukat/mapping/diffusion.py
@@ -467,9 +467,10 @@ def to_nifti(self, output_directory=os.getcwd(), base_file_name='Output',
                     nib.save(color_fa_nifti, base_path +
                              '_color_fa_map.nii.gz')
                 elif result == 'mask':
-                    mask_nifti = nib.Nifti1Image(self.mask.astype(np.uint16),
-                                                 affine=self.affine)
-                    nib.save(mask_nifti, base_path + '_mask.nii.gz')
+                    if self.mask is not None:
+                        mask_nifti = nib.Nifti1Image(self.mask.astype(np.uint16),
+                                                     affine=self.affine)
+                        nib.save(mask_nifti, base_path + '_mask.nii.gz')
         else:
             raise ValueError('No NIFTI file saved. The variable "maps" '
                              'should be "all" or a list of maps from '