Update package/samplers/hebo/sampler.py

Co-authored-by: Shuhei Watanabe <[email protected]>
optuna · Dec 8, 2024 · 4e1a019 · 4e1a019
1 parent a82754e
commit 4e1a019
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/package/samplers/hebo/sampler.py b/package/samplers/hebo/sampler.py
@@ -107,7 +107,8 @@ def _sample_relative_stateless(
         else:
             target_states = [TrialState.COMPLETE]
 
-        trials = study.get_trials(deepcopy=False, states=target_states)
+        use_cache = not self._constant_liar
+        trials = study._get_trials(deepcopy=False, states=states, use_cache=use_cache)
         if len([t for t in trials if t.state == TrialState.COMPLETE]) < 1:
             # note: The backend HEBO implementation uses Sobol sampling here.
             # This sampler does not call `hebo.suggest()` here because