tests

aramis-lab · Oct 3, 2024 · 299b7dd · 299b7dd
1 parent e34a034
commit 299b7dd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/clinicadl/trainer/config/train.py b/clinicadl/trainer/config/train.py
@@ -76,6 +76,7 @@ def __init__(self, **kwargs):
             optimization=kwargs,
             optimizer=kwargs,
             reproducibility=kwargs,
+            split=kwargs,
             ssda=kwargs,
             transfer_learning=kwargs,
             transforms=kwargs,
@@ -95,6 +96,7 @@ def _update(self, config_dict: Dict[str, Any]) -> None:
         self.optimization.__dict__.update(config_dict)
         self.optimizer.__dict__.update(config_dict)
         self.reproducibility.__dict__.update(config_dict)
+        self.split.__dict__.update(config_dict)
         self.ssda.__dict__.update(config_dict)
         self.transfer_learning.__dict__.update(config_dict)
         self.transforms.__dict__.update(config_dict)