fixed lits

kaiko-ai · Nov 28, 2024 · 55a5d20 · 55a5d20
1 parent 17f86be
commit 55a5d20
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/eva/vision/data/datasets/segmentation/lits.py b/src/eva/vision/data/datasets/segmentation/lits.py
@@ -105,8 +105,8 @@ def validate(self) -> None:
         _validators.check_dataset_integrity(
             self,
             length=self._expected_dataset_lengths.get(self._split, 0),
-            n_classes=2,
-            first_and_last_labels=("liver", "tumor"),
+            n_classes=3,
+            first_and_last_labels=("background", "tumor"),
         )
 
     @override