From 8d0e66ec207d3c39adeb0b152fad4b11eb1f7ace Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Mon, 27 May 2024 11:42:29 +0200
Subject: [PATCH] [segmentation] Fix typing

---
 src/mednet/libs/segmentation/models/typing.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/mednet/libs/segmentation/models/typing.py b/src/mednet/libs/segmentation/models/typing.py
index 5a24afb6..11ec1457 100644
--- a/src/mednet/libs/segmentation/models/typing.py
+++ b/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."""
-- 
GitLab