PrunedRelaxedClockModel is broken after meanRate is changed to Function

#15
BEAST2-Dev · Jul 29, 2021 · c88e042 · c88e042
1 parent 7cd6d32
commit c88e042
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/beast/evolution/branchratemodel/PrunedRelaxedClockModel.java b/src/beast/evolution/branchratemodel/PrunedRelaxedClockModel.java
@@ -3,6 +3,8 @@
 
 import beast.core.Description;
 import beast.core.Input;
+import beast.core.StateNode;
+import beast.core.parameter.RealParameter;
 import beast.evolution.tree.Node;
 import beast.evolution.tree.PrunedTree;
 import beast.evolution.tree.TreeInterface;
@@ -67,8 +69,11 @@ protected boolean requiresRecalculation() {
     }
 
     private boolean rq() {
+        if (! (rates.meanRate instanceof RealParameter) )
+            throw new UnsupportedOperationException("meanRate has to be RealParameter !");
+
         // before tree, since tree might not be valid if only rates changed
-        if( rates.distribution.isDirtyCalculation() || rates.meanRate.somethingIsDirty() ) {
+        if( rates.distribution.isDirtyCalculation() || ((RealParameter) rates.meanRate).somethingIsDirty() ) {
             return true;
         }