Merge pull request #1013 from nextstrain/fix-clade-color-ordering

Fix color assignment for clades
nextstrain · Oct 13, 2022 · 44ab71a · 44ab71a
2 parents 102e14e + c52433f
commit 44ab71a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/scripts/assign-colors.py b/scripts/assign-colors.py
@@ -35,6 +35,9 @@
     if args.metadata:
         metadata = pd.read_csv(args.metadata, delimiter='\t')
         for name, trait in assignment.items():
+            if name == "clade_membership" and "Nextstrain_clade" in metadata:
+                subset_present = [x for x in assignment[name] if x in metadata["Nextstrain_clade"].unique()]
+                assignment[name] = subset_present
             if name in metadata:
                 subset_present = [x for x in assignment[name] if x in metadata[name].unique()]
                 assignment[name] = subset_present