fix indexing error in beagle tree likellihood fixes #1168

CompEvol · Sep 23, 2024 · e6fd752 · e6fd752
1 parent c520aaf
commit e6fd752
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/beast/base/evolution/likelihood/BeagleTreeLikelihood.java b/src/beast/base/evolution/likelihood/BeagleTreeLikelihood.java
@@ -959,7 +959,7 @@ private int traverse(Node node, int[] operatorNumber, boolean flip) {
             matrixUpdateIndices[eigenIndex][updateCount] = matrixBufferHelper.getOffsetIndex(nodeNum);
 
             if (substitutionModel.canReturnComplexDiagonalization()) {
-                for (int i = 0; i < m_siteModel.getCategoryCount(); i++) {
+                for (int i = 0; i < this.categoryCount; i++) {
                     final double jointBranchRate = m_siteModel.getRateForCategory(i, node) * branchRate;
                     substitutionModel.getTransitionProbabilities(node, node.getParent().getHeight(), node.getHeight(), jointBranchRate, probabilities);
                     //System.out.println(node.getNr() + " " + Arrays.toString(m_fProbabilities));