Update package/samplers/hebo/sampler.py

Co-authored-by: Shuhei Watanabe <[email protected]>
optuna · Dec 10, 2024 · 1de364d · 1de364d
1 parent a46a02e
commit 1de364d
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/package/samplers/hebo/sampler.py b/package/samplers/hebo/sampler.py
@@ -126,10 +126,9 @@ def _sample_relative_stateless(
 
         seed = self._rng.randint((1 << 31) - 1)
         hebo = HEBO(self._convert_to_hebo_design_space(search_space), scramble_seed=seed)
-        valid_trials = [
+        params = pd.DataFrame([
             t.params for t in trials if all(name in trial.params for name in search_space)
-        ]
-        params = pd.DataFrame([t.params for t in valid_trials])
+        ])
         values = np.array(
             [
                 sign * t.value if t.state == TrialState.COMPLETE else worst_value