only generate gradient XML blocks when needed

beast-dev · Oct 10, 2024 · 68a519c · 68a519c
1 parent 9c9f62e
commit 68a519c
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/dr/app/beauti/generator/ClockModelGenerator.java b/src/dr/app/beauti/generator/ClockModelGenerator.java
@@ -66,6 +66,8 @@
 import dr.util.Attribute;
 import dr.xml.XMLParser;
 
+import java.util.Map;
+
 import static dr.inference.model.ParameterParser.PARAMETER;
 import static dr.inferencexml.distribution.PriorParsers.*;
 import static dr.inferencexml.distribution.shrinkage.BayesianBridgeLikelihoodParser.*;
@@ -299,10 +301,13 @@ public void writeBranchRatesModel(PartitionClockModel clockModel, XMLWriter writ
 
                 writeCovarianceStatistic(writer, tag, prefix, treePrefix);
 
-                //TODO put this in an IF statement depending on choice in BEAUti
+                //TODO add more String constants for this type of code
                 boolean generateScaleGradient = false;
-                if (options.getOperator("HMCRCS").isUsed()) {
-                    generateScaleGradient = true;
+
+                for (Operator operator : options.selectOperators()) {
+                    if (operator.getName().equals("HMC relaxed clock location and scale") && operator.isUsed()) {
+                        generateScaleGradient = true;
+                    }
                 }
 
                 if (generateScaleGradient) {