location scale gradient generator code was in the wrong place

beast-dev · Dec 24, 2024 · 41a001c · 41a001c
1 parent fd162f3
commit 41a001c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/dr/app/beauti/generator/ClockModelGenerator.java b/src/dr/app/beauti/generator/ClockModelGenerator.java
@@ -340,6 +340,9 @@ public void writeBranchRatesModel(PartitionClockModel clockModel, XMLWriter writ
                     writer.writeIDref(DefaultTreeModel.TREE_MODEL, treePrefix + DefaultTreeModel.TREE_MODEL);
                     writer.writeCloseTag(CTMCScalePriorParser.MODEL_NAME);
 
+                }
+
+                if (generateScaleGradient){
                     //location gradient
                     writer.writeOpenTag(LocationScaleGradientParser.NAME, new Attribute[]{
                             new Attribute.Default<>(XMLParser.ID, prefix + LocationGradient.LOCATION_GRADIENT),
@@ -352,9 +355,6 @@ public void writeBranchRatesModel(PartitionClockModel clockModel, XMLWriter writ
                     writer.writeCloseTag(LocationScaleGradientParser.LOCATION);
                     writer.writeCloseTag(LocationScaleGradientParser.NAME);
 
-                }
-
-                if (generateScaleGradient){
                     //scale gradient
                     writer.writeOpenTag(LocationScaleGradientParser.NAME, new Attribute[]{
                             new Attribute.Default<>(XMLParser.ID, prefix + ScaleGradient.SCALE_GRADIENT),