From f5b8a9952ed2f82f1099f69715696237d1e1e8a4 Mon Sep 17 00:00:00 2001 From: Terry Truong Date: Mon, 11 Jul 2022 20:16:31 +1000 Subject: Include picked-names when collapsing nodes in 'images' tree Without this, nodes like 'giant panda' were getting removed --- backend/tolData/genReducedTrees.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/tolData/genReducedTrees.py b/backend/tolData/genReducedTrees.py index a921be4..907aad9 100755 --- a/backend/tolData/genReducedTrees.py +++ b/backend/tolData/genReducedTrees.py @@ -98,7 +98,7 @@ def genImagesOnlyTree(dbCur, nodesWithImgOrPicked, pickedNames, rootName): removeCompositeNodes(nodeMap) print(f"Result has {len(nodeMap)} nodes") print("Removing 'collapsible' nodes") - removeCollapsibleNodes(nodeMap, {}) + removeCollapsibleNodes(nodeMap, pickedNames) print(f"Result has {len(nodeMap)} nodes") print(f"Updating 'tips' values") # Needed for next trimming step updateTips(rootName, nodeMap) -- cgit v1.2.3