diff --git a/examples/viz_optimizers.py b/examples/viz_optimizers.py index 5c1cba1..9d7f2e4 100644 --- a/examples/viz_optimizers.py +++ b/examples/viz_optimizers.py @@ -136,7 +136,7 @@ def execute_experiments( space=space, algo=tpe.suggest, max_evals=200, - rstate=np.random.RandomState(seed), + rstate=np.random.default_rng(seed), ) print(best["lr"], optimizer_class)