Handle markers right of last genotyped marker differently

szhan · Jan 14, 2024 · c316e8c · c316e8c
1 parent 3fc04e2
commit c316e8c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/tests/beagle_numba.py b/python/tests/beagle_numba.py
@@ -328,8 +328,8 @@ def interpolate_allele_prob(sm, ref_h, typed_pos, untyped_pos, typed_cm, untyped
             for a in alleles:
                 if ref_h[i, j] == a:
                     if k == m - 1:
-                        # TODO: Consult BEAGLE source code.
-                        pass
+                        # FIXME: It's unclear how BEAGLE handles this case.
+                        p[i, a] += sm[k, j]
                     else:
                         p[i, a] += w * sm[k, j]
                         p[i, a] += (1 - w) * sm[k + 1, j]