Update array_utils.py (#303)

Increase tolerance from 1e-5 (default) to 1e-4 when checking sum_to_one

Author: averyma

baal/utils/array_utils.py

@@ -15,7 +15,7 @@ def to_prob(probabilities: np.ndarray):
     """
     not_bounded = np.min(probabilities) < 0 or np.max(probabilities) > 1.0
     multiclass = probabilities.shape[1] > 1
-    sum_to_one = np.allclose(probabilities.sum(1), 1)
+    sum_to_one = np.allclose(probabilities.sum(1), 1, rtol=1e-4)
     if not_bounded or (multiclass and not sum_to_one):
         if multiclass:
             probabilities = softmax(probabilities, 1)