From af2c1d67ab98f37137636d90612382754533818a Mon Sep 17 00:00:00 2001 From: camillebrianceau Date: Wed, 9 Oct 2024 11:16:34 +0200 Subject: [PATCH] cli --- clinicadl/interpret/config.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/clinicadl/interpret/config.py b/clinicadl/interpret/config.py index ed8ad7a72..997b8aacd 100644 --- a/clinicadl/interpret/config.py +++ b/clinicadl/interpret/config.py @@ -7,7 +7,7 @@ from clinicadl.caps_dataset.data_config import DataConfig as DataBaseConfig from clinicadl.caps_dataset.dataloader_config import DataLoaderConfig from clinicadl.interpret.gradients import GradCam, Gradients, VanillaBackProp -from clinicadl.maps_manager.config import MapsManagerConfigBase +from clinicadl.maps_manager.config import MapsManagerConfig as MapsManagerConfigBase from clinicadl.maps_manager.maps_manager import MapsManager from clinicadl.predictor.validation import ValidationConfig from clinicadl.splitter.config import SplitConfig @@ -20,6 +20,8 @@ class MapsManagerConfig(MapsManagerConfigBase): + save_tensor: bool = False + def check_output_saving_tensor(self, network_task: str) -> None: # Check if task is reconstruction for "save_tensor" and "save_nifti" if self.save_tensor and network_task != "reconstruction":