Merge pull request #4023 from not522/fix-skopt-seed

Fix duplicated sampling of `SkoptSampler`
optuna · Oct 4, 2022 · 7da2f2d · 7da2f2d
2 parents 2498b7c + f54e1ef
commit 7da2f2d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/optuna/integration/skopt.py b/optuna/integration/skopt.py
@@ -136,6 +136,7 @@ def __init__(
 
         if seed is not None and "random_state" not in self._skopt_kwargs:
             self._skopt_kwargs["random_state"] = seed
+        self._rng: Optional[np.random.RandomState] = None
 
     def reseed_rng(self) -> None:
 
@@ -177,8 +178,12 @@ def sample_relative(
             return {}
 
         optimizer = _Optimizer(search_space, self._skopt_kwargs)
+        if self._rng is not None:
+            optimizer._optimizer.rng = self._rng
         optimizer.tell(study, complete_trials)
-        return optimizer.ask()
+        params = optimizer.ask()
+        self._rng = optimizer._optimizer.rng
+        return params
 
     def sample_independent(
         self,