tests

clinicadl/trainer/config/train.py

@@ -19,6 +19,7 @@
 from clinicadl.network.config import NetworkConfig
 from clinicadl.optimizer.optimization import OptimizationConfig
 from clinicadl.optimizer.optimizer import OptimizerConfig
+from clinicadl.splitter.config import SplitConfig
 from clinicadl.splitter.validation import ValidationConfig
 from clinicadl.trainer.transfer_learning import TransferLearningConfig
 from clinicadl.transforms.config import TransformsConfig
@@ -48,6 +49,7 @@ class TrainConfig(BaseModel, ABC):
     optimization: OptimizationConfig
     optimizer: OptimizerConfig
     reproducibility: ReproducibilityConfig
+    split: SplitConfig
     ssda: SSDAConfig
     transfer_learning: TransferLearningConfig
     transforms: TransformsConfig

tests/unittests/train/trainer/test_training_config.py

@@ -181,14 +181,14 @@ def test_fails_validations(bad_inputs, training_config):
 def test_passes_validations(good_inputs, training_config):
     c = training_config(**good_inputs)
     assert not c.computational.gpu
-    assert c.validation.n_splits == 7
+    assert c.split.n_splits == 7
     assert c.optimizer.optimizer == "Adagrad"
     assert c.transforms.data_augmentation == ("Smoothing",)
     assert c.data.diagnoses == ("AD",)
     assert c.dataloader.batch_size == 1
     assert c.transforms.size_reduction_factor == 5
     assert c.optimizer.learning_rate == 1e-1
-    assert c.validation.split == (0,)
+    assert c.split.split == (0,)
     assert c.early_stopping.tolerance == 0.0