aboutsummaryrefslogtreecommitdiff
path: root/backend/tolData/genReducedTrees.py
diff options
context:
space:
mode:
authorTerry Truong <terry06890@gmail.com>2022-07-11 20:16:31 +1000
committerTerry Truong <terry06890@gmail.com>2022-07-11 20:16:31 +1000
commitf5b8a9952ed2f82f1099f69715696237d1e1e8a4 (patch)
treed1cdd3009fe9c1f534fd00a76a397f0c84b39201 /backend/tolData/genReducedTrees.py
parent6231aa2dfef46baecbb68b199196317592607651 (diff)
Include picked-names when collapsing nodes in 'images' tree
Without this, nodes like 'giant panda' were getting removed
Diffstat (limited to 'backend/tolData/genReducedTrees.py')
-rwxr-xr-xbackend/tolData/genReducedTrees.py2
1 files changed, 1 insertions, 1 deletions
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)