diff --git a/tests/test_careamist.py b/tests/test_careamist.py index 9b0040249..88b253f91 100644 --- a/tests/test_careamist.py +++ b/tests/test_careamist.py @@ -1144,6 +1144,7 @@ def test_all_parameters_used(tmp_path, minimum_n2v_configuration): config.training_config.num_epochs = 2 config.data_config.batch_size = 2 config.data_config.patch_size = (16, 16) + config.data_config.dataloader_params = {"num_workers": 2} # TODO add other params ? careamist = CAREamist(config, work_dir=tmp_path) @@ -1151,3 +1152,4 @@ def test_all_parameters_used(tmp_path, minimum_n2v_configuration): assert careamist.cfg.training_config.num_epochs == 2 assert careamist.cfg.data_config.batch_size == 2 assert all(i == 16 for i in careamist.cfg.data_config.patch_size) + assert careamist.cfg.data_config.dataloader_params["num_workers"] == 2