From 3fdd740f1f1813306348c1709ee84c19b436bbc6 Mon Sep 17 00:00:00 2001 From: camillebrianceau Date: Thu, 24 Oct 2024 13:30:45 +0200 Subject: [PATCH] adapt without preprocessing --- clinicadl/caps_dataset/extraction/config.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/clinicadl/caps_dataset/extraction/config.py b/clinicadl/caps_dataset/extraction/config.py index bb4bed89a..007df568d 100644 --- a/clinicadl/caps_dataset/extraction/config.py +++ b/clinicadl/caps_dataset/extraction/config.py @@ -58,7 +58,7 @@ class ExtractionSliceConfig(ExtractionConfig): slice_direction: SliceDirection = SliceDirection.SAGITTAL slice_mode: SliceMode = SliceMode.RGB num_slices: Optional[NonNegativeInt] = None - discarded_slices: Union[int, tuple] = (0,) + discarded_slices: Union[int, Tuple] = (0,) extract_method: ExtractionMethod = ExtractionMethod.SLICE @field_validator("slice_direction", mode="before") @@ -67,7 +67,7 @@ def check_slice_direction(cls, v: str): return SliceDirection(str(v)) @field_validator("discarded_slices", mode="before") - def compute_discarded_slice(cls, v: Union[int, tuple]) -> Tuple[int, int]: + def compute_discarded_slice(cls, v: Union[int, Tuple]) -> Tuple[int, int]: return compute_discarded_slices(v)