[segmentation] Fix typing

src/mednet/libs/segmentation/models/typing.py

@@ -9,5 +9,7 @@ import typing
 Checkpoint: typing.TypeAlias = typing.MutableMapping[str, typing.Any]
 """Definition of a lightning checkpoint."""
 
-SegmentationPrediction: typing.TypeAlias = tuple[pathlib.Path, pathlib.Path]
-"""The sample name, the target, and the predicted value."""
+SegmentationPrediction: typing.TypeAlias = tuple[
+    pathlib.Path, pathlib.Path, pathlib.Path, pathlib.Path
+]
+"""The sample name, the target, mask, and the prediction."""