splitter in init trainer

aramis-lab · Oct 14, 2024 · 6f6933a · 6f6933a
1 parent 825f5fd
commit 6f6933a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/clinicadl/predictor/predictor.py b/clinicadl/predictor/predictor.py
@@ -53,7 +53,11 @@ def __init__(self, _config: Union[PredictConfig, InterpretConfig]) -> None:
 
         self.maps_manager = MapsManager(_config.maps_manager.maps_dir)
         self._config.adapt_with_maps_manager_info(self.maps_manager)
-        print(self._config.data.model_dump())
+        print(
+            self._config.data.model_dump(
+                exclude=set(["preprocessing_dict", "mode", "caps_dict"])
+            )
+        )
         tmp = self._config.data.model_dump(
             exclude=set(["preprocessing_dict", "mode", "caps_dict"])
         )