Merge pull request #309 from nspope/fix-mutation-count

Fix excess mutation count at last edge (closes #308)
tskit-dev · Aug 20, 2023 · 9e17948 · 9e17948
2 parents 41c2689 + ff27b0d
commit 9e17948
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tsdate/core.py b/tsdate/core.py
@@ -130,7 +130,8 @@ def get_mut_edges(ts):
         """
         mut_edges = np.zeros(ts.num_edges, dtype=np.int64)
         for m in ts.mutations():
-            mut_edges[m.edge] += 1
+            if m.edge != tskit.NULL:
+                mut_edges[m.edge] += 1
         return mut_edges
 
     @staticmethod