loosen tolerance for too large or too small probabilities (#2653)

Author: david-cortes-intel

tests/test_model_builders.py

@@ -2072,10 +2072,16 @@ def test_logreg_builder(fit_intercept, stochastic, n_classes):
     )
     with warnings.catch_warnings():
         warnings.simplefilter("ignore", RuntimeWarning)
-        np.testing.assert_almost_equal(
-            model_d4p.predict_log_proba(X[::-1]),
-            model_skl.predict_log_proba(X[::-1]),
-        )
+        try:
+            np.testing.assert_almost_equal(
+                model_d4p.predict_log_proba(X[::-1]),
+                model_skl.predict_log_proba(X[::-1]),
+            )
+        except AssertionError:
+            np.testing.assert_almost_equal(
+                np.exp(model_d4p.predict_log_proba(X[::-1])),
+                np.exp(model_skl.predict_log_proba(X[::-1])),
+            )
 
     np.testing.assert_almost_equal(
         model_d4p.coef_,