Skip to content

Commit

Permalink
Update package/samplers/hebo/sampler.py
Browse files Browse the repository at this point in the history
Co-authored-by: Shuhei Watanabe <[email protected]>
  • Loading branch information
eukaryo and nabenabe0928 authored Dec 8, 2024
1 parent d9c00b2 commit a82754e
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion package/samplers/hebo/sampler.py
Original file line number Diff line number Diff line change
Expand Up @@ -121,7 +121,8 @@ def _sample_relative_stateless(
worst_value = min(t.value for t in trials if t.state == TrialState.COMPLETE)
sign = 1 if study.direction == StudyDirection.MINIMIZE else -1

hebo = HEBO(self._convert_to_hebo_design_space(search_space), scramble_seed=self._seed)
seed = self._rng.randint((1 << 31) - 1)
hebo = HEBO(self._convert_to_hebo_design_space(search_space), scramble_seed=seed)
for t in trials:
if t.state == TrialState.COMPLETE:
hebo_params = {name: t.params[name] for name in search_space.keys()}
Expand Down

0 comments on commit a82754e

Please sign in to comment.