diff --git a/test/tuner_test.py b/test/tuner_test.py index 131fa3971f..139a3f4844 100644 --- a/test/tuner_test.py +++ b/test/tuner_test.py @@ -49,6 +49,10 @@ def switch(dispatch_type, dispatch_name): 'optimize_mode': 'maximize' } } + if dispatch_name == 'BatchTuner': + experiment_config['searchSpacePath'] = 'batchtuner_search_space.json' + else: + experiment_config['searchSpacePath'] = 'search_space.json' dump_yml_content(config_path, experiment_config) def test_builtin_dispatcher(dispatch_type, dispatch_name): diff --git a/test/tuner_test/batchtuner_search_space.json b/test/tuner_test/batchtuner_search_space.json new file mode 100644 index 0000000000..5d3beeee85 --- /dev/null +++ b/test/tuner_test/batchtuner_search_space.json @@ -0,0 +1,8 @@ +{ + "combine_params": + { + "_type" : "choice", + "_value" : [{"x": 1}, + {"x": 100}] + } +}