More agressive replacement of RandomState with default_rng

Signed-off-by: Avi Shinnar <[email protected]>
IBM · Aug 22, 2024 · 626df14 · 626df14
1 parent e237f5b
commit 626df14
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/lale/lib/lale/smac.py b/lale/lib/lale/smac.py
@@ -172,7 +172,7 @@ def f(trainable):
         try:
             smac = orig_SMAC(
                 scenario=self.scenario,
-                rng=np.random.RandomState(42),
+                rng=np.random.default_rng(42),
                 tae_runner=lale_op_smac_tae(self.estimator, f),
             )
             incumbent = smac.optimize()

diff --git a/test/test_optimizers.py b/test/test_optimizers.py
@@ -124,7 +124,7 @@ def test_smac(self):
         tae = iris_fmin_tae(lr, num_folds=2)
         print("Optimizing! Depending on your machine, this might take a few minutes.")
         smac = orig_SMAC(
-            scenario=scenario, rng=np.random.RandomState(42), tae_runner=tae
+            scenario=scenario, rng=np.random.default_rng(42), tae_runner=tae
         )
 
         incumbent = smac.optimize()
@@ -167,7 +167,7 @@ def dont_test_smac_choice(self):
         tae = iris_fmin_tae(planned_pipeline1, num_folds=2)
         print("Optimizing! Depending on your machine, this might take a few minutes.")
         smac = orig_SMAC(
-            scenario=scenario, rng=np.random.RandomState(42), tae_runner=tae
+            scenario=scenario, rng=np.random.default_rng(42), tae_runner=tae
         )
 
         incumbent = smac.optimize()