From f10aefbc7eeacc36d35d890da9ef1dc7f53adea5 Mon Sep 17 00:00:00 2001 From: Gerrit Birkeland Date: Sun, 27 Apr 2025 09:19:01 -0600 Subject: [PATCH] Use ensure_rng in AcquisitionFunction class --- bayes_opt/acquisition.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/bayes_opt/acquisition.py b/bayes_opt/acquisition.py index a81a6df13..93b2f3ede 100644 --- a/bayes_opt/acquisition.py +++ b/bayes_opt/acquisition.py @@ -38,6 +38,7 @@ TargetSpaceEmptyError, ) from bayes_opt.target_space import TargetSpace +from bayes_opt.util import ensure_rng if TYPE_CHECKING: from collections.abc import Callable, Sequence @@ -60,13 +61,7 @@ class AcquisitionFunction(abc.ABC): """ def __init__(self, random_state: int | RandomState | None = None) -> None: - if random_state is not None: - if isinstance(random_state, RandomState): - self.random_state = random_state - else: - self.random_state = RandomState(random_state) - else: - self.random_state = RandomState() + self.random_state = ensure_rng(random_state) self.i = 0 def _serialize_random_state(self) -> dict | None: