diff --git a/nnunetv2/imageio/nibabel_reader_writer.py b/nnunetv2/imageio/nibabel_reader_writer.py index 78fb17ac1..2854da4b5 100644 --- a/nnunetv2/imageio/nibabel_reader_writer.py +++ b/nnunetv2/imageio/nibabel_reader_writer.py @@ -31,8 +31,6 @@ class NibabelIO(BaseReaderWriter): supported_file_endings = [ '.nii', '.nii.gz', - '.nrrd', - '.mha' ] def read_images(self, image_fnames: Union[List[str], Tuple[str, ...]]) -> Tuple[np.ndarray, dict]: @@ -110,8 +108,6 @@ class NibabelIOWithReorient(BaseReaderWriter): supported_file_endings = [ '.nii', '.nii.gz', - '.nrrd', - '.mha' ] def read_images(self, image_fnames: Union[List[str], Tuple[str, ...]]) -> Tuple[np.ndarray, dict]: