diff --git a/tests/test_pyglmnet.py b/tests/test_pyglmnet.py index 43a098f1..66724bdd 100644 --- a/tests/test_pyglmnet.py +++ b/tests/test_pyglmnet.py @@ -415,12 +415,11 @@ def test_random_state_consistency(): ypred_a = glm_a.fit_predict(Xtrain, ytrain) glm_b = GLM(distr="gaussian", random_state=1) ypred_b = glm_b.fit_predict(Xtrain, ytrain) - ypred_c = glm_b.fit_predict(Xtrain, ytrain) + glm_a.fit(Xtrain, ytrain) + glm_b.fit(Xtrain, ytrain) # Consistency between two different models assert_array_equal(ypred_a, ypred_b) - # Consistency between different run of the same model - assert_array_equal(ypred_b, ypred_c) # Test also cross-validation glm_cv_a = GLMCV(distr="gaussian", cv=3, random_state=1)