From 9445adcc6e8aace37bea2835d1152f500029ec32 Mon Sep 17 00:00:00 2001 From: xadupre Date: Mon, 23 Dec 2024 10:38:49 +0100 Subject: [PATCH] atol Signed-off-by: xadupre --- tests/xgboost/test_xgboost_converters.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/xgboost/test_xgboost_converters.py b/tests/xgboost/test_xgboost_converters.py index 44cf71f0..7eec135c 100644 --- a/tests/xgboost/test_xgboost_converters.py +++ b/tests/xgboost/test_xgboost_converters.py @@ -653,12 +653,12 @@ def test_xgb_cost(self): y = oinf.run(None, {"X": X_test.astype(np.float32)}) if cl == XGBRegressor: exp = clr.predict(X_test) - assert_almost_equal(exp, y[0].ravel(), decimal=5) + assert_almost_equal(exp, y[0].ravel(), decimal=4) else: if "softmax" not in obj: exp = clr.predict_proba(X_test) got = pandas.DataFrame(y[1]).values - assert_almost_equal(exp, got, decimal=5) + assert_almost_equal(exp, got, decimal=4) exp = clr.predict(X_test[:10]) assert_almost_equal(exp, y[0][:10])