Take sum of probs

Author: szhan

python/tests/beagle_numba.py

@@ -566,8 +566,8 @@ def interpolate_allele_probs(
                 # Threshold based on "the number of subsets in the partition Am of H".
                 threshold_Am = np.sum(is_a_in_ref_h)
                 _MIN_THRESHOLD = min(0.005, 1 / threshold_Am)
-            sum_probs_a_k = state_mat[k, is_a_in_ref_h]
-            sum_probs_a_kM1 = state_mat[k - 1, is_a_in_ref_h]
+            sum_probs_a_k = np.sum(state_mat[k, is_a_in_ref_h])
+            sum_probs_a_kM1 = np.sum(state_mat[k - 1, is_a_in_ref_h])
             if k == 0:
                 # See 'setFirstAlleleProbs' in 'LSHapBaum.java'.
                 assert w == 1.0, "Weight should be 1.0."