Update package/samplers/hebo_base_sampler/sampler.py

Co-authored-by: Shuhei Watanabe <[email protected]>
optuna · Dec 4, 2024 · 1f28f09 · 1f28f09
1 parent 11f1cff
commit 1f28f09
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/package/samplers/hebo_base_sampler/sampler.py b/package/samplers/hebo_base_sampler/sampler.py
@@ -132,7 +132,8 @@ def _convert_to_hebo_design_space(
         return DesignSpace().parse(design_space)
 
     def infer_relative_search_space(self, study, trial):  # type: ignore
-        return optuna.search_space.intersection_search_space(study.get_trials(deepcopy=False))
+        return optuna.search_space.intersection_search_space(study._get_trials(deepcopy=False, use_cache=True))
+
 
     def sample_independent(self, study, trial, param_name, param_distribution):  # type: ignore
         return self._independent_sampler.sample_independent(