diff --git a/package/samplers/hebo/sampler.py b/package/samplers/hebo/sampler.py index 7b0a3d15..641fd39f 100644 --- 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