Merge pull request #297 from nextstrain/phylo-ancestral-root

phylogenetic: Add config option `ancestral_root_seq`

Author: joverlee521

phylogenetic/build-configs/ci/config.yaml

@@ -2,6 +2,7 @@ custom_rules:
   - build-configs/ci/copy_example_data.smk
 
 reference: "defaults/reference.fasta"
+ancestral_root_seq: "defaults/reference.fasta"
 genome_annotation: "defaults/genome_annotation.gff3"
 genbank_reference: "defaults/reference.gb"
 include: "defaults/hmpxv1/include.txt"

phylogenetic/rules/annotate_phylogeny.smk

@@ -31,13 +31,19 @@ rule ancestral:
         node_data=build_dir + "/{build_name}/nt_muts.json",
     params:
         inference="joint",
+        root_sequence=lambda w: (
+            f"--root-sequence {config['ancestral_root_seq']!r}"
+            if config.get("ancestral_root_seq")
+            else ""
+        ),
     shell:
         """
         augur ancestral \
             --tree {input.tree} \
             --alignment {input.alignment} \
             --output-node-data {output.node_data} \
-            --inference {params.inference}
+            --inference {params.inference} \
+            {params.root_sequence}
         """