relax tolerance for the second place comparing intercept

NVIDIA · Dec 10, 2024 · acdcd2c · acdcd2c
1 parent 19b2970
commit acdcd2c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tests/test_logistic_regression.py b/python/tests/test_logistic_regression.py
@@ -589,7 +589,7 @@ def test_compat(
             tolerance,
         )
         assert isinstance(blor_model.interceptVector, DenseVector)
-        assert array_equal(blor_model.interceptVector.toArray(), [0.0])
+        assert array_equal(blor_model.interceptVector.toArray(), [0.0], tolerance)
 
         example = bdf.head()
         if example: