Skip to content

Commit 9a464f2

Browse files
authored
Use ensure_rng in AcquisitionFunction class (#561)
1 parent 1714504 commit 9a464f2

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

bayes_opt/acquisition.py

+2-7
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@
3838
TargetSpaceEmptyError,
3939
)
4040
from bayes_opt.target_space import TargetSpace
41+
from bayes_opt.util import ensure_rng
4142

4243
if TYPE_CHECKING:
4344
from collections.abc import Callable, Sequence
@@ -60,13 +61,7 @@ class AcquisitionFunction(abc.ABC):
6061
"""
6162

6263
def __init__(self, random_state: int | RandomState | None = None) -> None:
63-
if random_state is not None:
64-
if isinstance(random_state, RandomState):
65-
self.random_state = random_state
66-
else:
67-
self.random_state = RandomState(random_state)
68-
else:
69-
self.random_state = RandomState()
64+
self.random_state = ensure_rng(random_state)
7065
self.i = 0
7166

7267
def _serialize_random_state(self) -> dict | None:

0 commit comments

Comments
 (0)