diff --git a/test/test_modules/test_pre_processing.py b/test/test_modules/test_pre_processing.py index 9d1feb03ac71b980f6a4cd1b0e6cac2a52d9625b..9b095bdafa648289a0620af9cdbe33883318b556 100644 --- a/test/test_modules/test_pre_processing.py +++ b/test/test_modules/test_pre_processing.py @@ -54,7 +54,8 @@ class TestPreProcessing: assert obj_with_exp_setup.data_store.search_name("generator") == [] obj_with_exp_setup.split_train_val_test() data_store = obj_with_exp_setup.data_store - assert data_store.search_scope("general.train") == sorted(["generator", "start", "end", "stations"]) + expected_params = ["generator", "start", "end", "stations", "permute_data"] + assert data_store.search_scope("general.train") == sorted(expected_params) assert data_store.search_name("generator") == sorted(["general.train", "general.val", "general.test", "general.train_val"])