pb with predict and interpret

aramis-lab · Oct 3, 2024 · af04e3f · af04e3f
1 parent df78384
commit af04e3f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/clinicadl/interpret/config.py b/clinicadl/interpret/config.py
@@ -8,6 +8,7 @@
 from clinicadl.caps_dataset.dataloader_config import DataLoaderConfig
 from clinicadl.interpret.gradients import GradCam, Gradients, VanillaBackProp
 from clinicadl.maps_manager.config import MapsManagerConfig
+from clinicadl.splitter.config import SplitConfig
 from clinicadl.splitter.validation import ValidationConfig
 from clinicadl.utils.computational.computational import ComputationalConfig
 from clinicadl.utils.enum import InterpretationMethod
@@ -50,5 +51,6 @@ class InterpretConfig(
     ValidationConfig,
     ComputationalConfig,
     DataLoaderConfig,
+    SplitConfig,
 ):
     """Config class to perform Transfer Learning."""
diff --git a/clinicadl/predict/config.py b/clinicadl/predict/config.py
@@ -5,6 +5,7 @@
 from clinicadl.maps_manager.config import (
     MapsManagerConfig as MapsManagerBaseConfig,
 )
+from clinicadl.splitter.config import SplitConfig
 from clinicadl.splitter.validation import ValidationConfig
 from clinicadl.utils.computational.computational import ComputationalConfig
 from clinicadl.utils.exceptions import ClinicaDLArgumentError  # type: ignore
@@ -34,5 +35,6 @@ class PredictConfig(
     ValidationConfig,
     ComputationalConfig,
     DataLoaderConfig,
+    SplitConfig,
 ):
     """Config class to perform Transfer Learning."""