From 63656cefdab35546d101255b47919af81ba76e90 Mon Sep 17 00:00:00 2001 From: ViralVerity Date: Mon, 14 Sep 2020 10:36:39 +0100 Subject: [PATCH] fixing display name error --- civet/scripts/make_tree_figures.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/civet/scripts/make_tree_figures.py b/civet/scripts/make_tree_figures.py index ca87edfe..7b39be31 100644 --- a/civet/scripts/make_tree_figures.py +++ b/civet/scripts/make_tree_figures.py @@ -69,7 +69,7 @@ def display_name(tree, tree_name, tree_dir, full_taxon_dict, query_dict, custom_ for k in tree.Objects: if k.branchType == 'leaf': name = k.name - + display_name = "" if "inserted" in name: collapsed_node_info, number_nodes = summarise_collapsed_node_for_label(tree_dir, name, tree_name, full_taxon_dict) k.traits["display"] = collapsed_node_info @@ -94,7 +94,8 @@ def display_name(tree, tree_name, tree_dir, full_taxon_dict, query_dict, custom_ else: display_name = display_name + "|" + taxon_obj.attribute_dict[label_element] count += 1 - k.traits["display"] = display_name + + k.traits["display"] = display_name k.node_number = 1 @@ -102,8 +103,8 @@ def display_name(tree, tree_name, tree_dir, full_taxon_dict, query_dict, custom_ else: if name.startswith("subtree"): number = name.split("_")[-1] - display = f"Tree {number}" - k.traits["display"] = display + display_name = f"Tree {number}" + k.traits["display"] = display_name k.node_number = 1 else: k.traits["display"] = name + "|" + "not in dict"