diff --git a/pkg/suggestion/v1beta1/skopt/base_service.py b/pkg/suggestion/v1beta1/skopt/base_service.py index e0203ee57e7..77140962d4e 100644 --- a/pkg/suggestion/v1beta1/skopt/base_service.py +++ b/pkg/suggestion/v1beta1/skopt/base_service.py @@ -118,11 +118,11 @@ def getSuggestions(self, trials, request_number): return_trial_list = [] - for i in range(request_number): - skopt_suggested = self.skopt_optimizer.ask() - logger.info("New suggested parameters for Trial: {}".format(skopt_suggested)) + skopt_suggested = self.skopt_optimizer.ask(n_points=request_number) + for suggestion in skopt_suggested: + logger.info("New suggested parameters for Trial: {}".format(suggestion)) return_trial_list.append( - BaseSkoptService.convert(self.search_space, skopt_suggested)) + BaseSkoptService.convert(self.search_space, suggestion)) logger.info("GetSuggestions return {} new Trials\n\n".format(request_number)) return return_trial_list