From 783919583259929d7f2f65fb00fc1644290d1ae1 Mon Sep 17 00:00:00 2001 From: Hiroki Takizawa Date: Wed, 4 Dec 2024 11:38:03 +0900 Subject: [PATCH] Update sampler.py --- package/samplers/hebo_base_sampler/sampler.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/package/samplers/hebo_base_sampler/sampler.py b/package/samplers/hebo_base_sampler/sampler.py index c415261d..0fe55356 100644 --- a/package/samplers/hebo_base_sampler/sampler.py +++ b/package/samplers/hebo_base_sampler/sampler.py @@ -182,9 +182,11 @@ def _convert_to_hebo_design_space( return DesignSpace().parse(design_space) def infer_relative_search_space( - self, study: Study, trial: FrozenTrial - ) -> dict[str, BaseDistribution]: - return optuna.search_space.intersection_search_space(study._get_trials(deepcopy=False, use_cache=True)) + self, study: Study, trial: FrozenTrial + ) -> dict[str, BaseDistribution]: + return optuna.search_space.intersection_search_space( + study._get_trials(deepcopy=False, use_cache=True) + ) def sample_independent( self,