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":