From b714476ffce436d455e3723e1fa00a7246ac1c3b Mon Sep 17 00:00:00 2001 From: Hiroki Takizawa Date: Wed, 4 Dec 2024 13:28:42 +0900 Subject: [PATCH] Update sampler.py --- package/samplers/hebo_base_sampler/sampler.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/package/samplers/hebo_base_sampler/sampler.py b/package/samplers/hebo_base_sampler/sampler.py index 2a062cd9..a73e16b2 100644 --- a/package/samplers/hebo_base_sampler/sampler.py +++ b/package/samplers/hebo_base_sampler/sampler.py @@ -108,13 +108,18 @@ def sample_relative( self._convert_to_hebo_design_space(search_space), scramble_seed=self._seed ) for t in trials: - hebo_params = {name: t.params[name] for name in search_space.keys()} if t.state == TrialState.COMPLETE: + hebo_params = {name: t.params[name] for name in search_space.keys()} hebo.observe( pd.DataFrame([hebo_params]), - np.asarray([map(lambda x: x * sign, t.values)]), + np.asarray([x * sign for x in t.values]), ) elif t.state == TrialState.RUNNING: + try: + hebo_params = {name: t.params[name] for name in search_space.keys()} + except: + # There are params which is not suggested yet. + continue # If `constant_liar == True`, assume that the RUNNING params result in bad values, # thus preventing the simultaneous suggestion of (almost) the same params # during parallel execution.