diff --git a/bayesian_optimization/bayesian_optimization.py b/bayesian_optimization/bayesian_optimization.py index 60a33eb..a2bf375 100644 --- a/bayesian_optimization/bayesian_optimization.py +++ b/bayesian_optimization/bayesian_optimization.py @@ -333,7 +333,7 @@ def __init__(self, interleaved_runs=2, *args, **kwargs): *args, **kwargs) for run in range(interleaved_runs)] self.rembos = cycle(self.rembos) - self.current_rembo = self.rembos.next() + self.current_rembo = next(self.rembos) self.X_ = [] self.y_ = [] @@ -361,4 +361,4 @@ def update(self, X, y): self.y_.append(y) self.current_rembo.update(X, y) - self.current_rembo = self.rembos.next() + self.current_rembo = next(self.rembos)