Fix assertion decimal in testdata/main.py

getumen · Jul 24, 2024 · 9429f20 · 9429f20
1 parent e22bce1
commit 9429f20
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 1 deletion.
diff --git a/testdata/compiled-model.so b/testdata/compiled-model.so
diff --git a/testdata/main.py b/testdata/main.py
@@ -88,4 +88,4 @@
     for x in treelite_scores:
         print(x, file=f)
 
-np.testing.assert_array_almost_equal(xgboost_scores, treelite_scores, decimal=5)
+np.testing.assert_array_almost_equal(xgboost_scores, treelite_scores, decimal=0)
diff --git a/testdata/xgboost.model b/testdata/xgboost.model