diff --git a/h2o-py/tests/testdir_algos/glm/pyunit_benign_glm.py b/h2o-py/tests/testdir_algos/glm/pyunit_benign_glm.py index d1ae565b4aa3..f18aadf54225 100644 --- a/h2o-py/tests/testdir_algos/glm/pyunit_benign_glm.py +++ b/h2o-py/tests/testdir_algos/glm/pyunit_benign_glm.py @@ -7,11 +7,10 @@ def test_benign(): training_data = h2o.import_file(pyunit_utils.locate("smalldata/logreg/benign.csv")) - + training_data[3] = training_data[3].asfactor() Y = 3 X = [0, 1, 2, 4, 5, 6, 7, 8, 9, 10] - model = H2OGeneralizedLinearEstimator(family="binomial", alpha=0, Lambda=1e-5) model = H2OGeneralizedLinearEstimator(family="binomial", alpha=0, lambda_=1e-5) model.train(x=X, y=Y, training_frame=training_data)