set fraction of rows that need to be almost equal to 0.9999

Signed-off-by: Jan-Benedikt Jagusch <jan.jagusch@gmail.com>
onnx · Aug 3, 2021 · 51988cb · 51988cb
1 parent 041beee
commit 51988cb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/lightgbm/test_objective_functions.py b/tests/lightgbm/test_objective_functions.py
@@ -14,6 +14,7 @@
 _N_ROWS=10_000
 _N_COLS=10
 _N_DECIMALS=5
+_FRAC = 0.9999
 
 _X = pd.DataFrame(np.random.random(size=(_N_ROWS, _N_COLS)))
 _Y = pd.Series(np.random.random(size=_N_ROWS))
@@ -83,4 +84,5 @@ def test_objective(self):
                     y_pred,
                     y_pred_onnx,
                     decimal=_N_DECIMALS,
+                    frac=_FRAC,
                 )