From aff2155ed978a2220af1c3e689f9257aa326a47c Mon Sep 17 00:00:00 2001 From: "Thomas J. Fan" Date: Wed, 24 Feb 2021 11:28:54 -0500 Subject: [PATCH 1/2] TST Fixes random seed for flaky test --- tests/model_selection/test_keras.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/tests/model_selection/test_keras.py b/tests/model_selection/test_keras.py index 7ea61b4ab..823f021bc 100644 --- a/tests/model_selection/test_keras.py +++ b/tests/model_selection/test_keras.py @@ -46,7 +46,9 @@ def _keras_build_fn(lr=0.01): @gen_cluster(client=True, Worker=Nanny, timeout=20) def test_keras(c, s, a, b): # Mirror the mnist dataset - X, y = make_classification(n_classes=10, n_features=784, n_informative=100) + X, y = make_classification( + n_classes=10, n_features=784, n_informative=100, random_state=0 + ) X = X.astype("float32") assert y.dtype == np.dtype("int64") @@ -56,7 +58,12 @@ def test_keras(c, s, a, b): params = {"lr": loguniform(1e-3, 1e-1)} search = IncrementalSearchCV( - model, params, max_iter=3, n_initial_parameters=5, decay_rate=None + model, + params, + max_iter=3, + n_initial_parameters=5, + decay_rate=None, + random_state=0, ) yield search.fit(X, y) # search.fit(X, y) From ce07328d11fdb4160cd171f06257ef8196dc019c Mon Sep 17 00:00:00 2001 From: "Thomas J. Fan" Date: Sat, 27 Feb 2021 10:44:34 -0500 Subject: [PATCH 2/2] Update tests/model_selection/test_keras.py Co-authored-by: Scott Sievert --- tests/model_selection/test_keras.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/model_selection/test_keras.py b/tests/model_selection/test_keras.py index 823f021bc..8b74307b0 100644 --- a/tests/model_selection/test_keras.py +++ b/tests/model_selection/test_keras.py @@ -55,7 +55,7 @@ def test_keras(c, s, a, b): model = KerasClassifier( model=_keras_build_fn, lr=0.01, verbose=False, loss="categorical_crossentropy", ) - params = {"lr": loguniform(1e-3, 1e-1)} + params = {"lr": loguniform(1e-3, 1e-1), "random_state": [1]} search = IncrementalSearchCV( model,