test fixed

arminwitte · Sep 15, 2023 · 99be4c8 · 99be4c8
1 parent 36d7a4d
commit 99be4c8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_prostate.py b/tests/test_prostate.py
@@ -74,8 +74,8 @@ def test_housing_gradientboostedtree():
         cart_settings={"lambda_l1":1.,"lambda_l2":1., "method":"regression:regularized"}
     )
     gbt.train(20)
-    p = c.predict(df_prostate[~train])
-    val = c.validate(df_prostate[~train])
+    p = gbt.predict(df_prostate[~train])
+    val = gbt.validate(df_prostate[~train])
     acc = val["R_squared"]
     np.testing.assert_allclose(
         p[:10],