Minor correction to unsupported file extension error message (#1177)

ivadomed · Jul 8, 2022 · 3099fdf · 3099fdf
1 parent a588984
commit 3099fdf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ivadomed/loader/segmentation_pair.py b/ivadomed/loader/segmentation_pair.py
@@ -281,7 +281,7 @@ def read_file(self, filename, is_gt=False):
         extension = imed_loader_utils.get_file_extension(filename)
         # TODO: remove "ome" from condition when implementing OMETIFF support (#739)
         if (not extension) or ("ome" in extension):
-            raise RuntimeError(f"The input file extension '{extension}' of '{Path(filename).stem}' is not "
+            raise RuntimeError(f"The input file extension '{extension}' of '{Path(filename).name}' is not "
                                f"supported. ivadomed supports the following "
                                f"file extensions: '.nii', '.nii.gz', '.png', '.tif', '.tiff', '.jpg' and '.jpeg'.")