atol

Signed-off-by: xadupre <[email protected]>
onnx · Dec 23, 2024 · 9445adc · 9445adc
1 parent 2630821
commit 9445adc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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])