diff options
Diffstat (limited to 'backend/data/genReducedTrees.py')
| -rwxr-xr-x | backend/data/genReducedTrees.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/backend/data/genReducedTrees.py b/backend/data/genReducedTrees.py index 17a66f2..9ffd007 100755 --- a/backend/data/genReducedTrees.py +++ b/backend/data/genReducedTrees.py @@ -53,7 +53,7 @@ def genPickedNodeTree(dbCur, pickedNames, rootName): print("Removing composite nodes") removedNames = removeCompositeNodes(nodeMap) print(f"Result has {len(nodeMap)} nodes") - print("Removing 'chain collapsible' nodes") + print("Removing 'collapsible' nodes") temp = removeCollapsibleNodes(nodeMap, pickedNames) removedNames.update(temp) print(f"Result has {len(nodeMap)} nodes") @@ -95,6 +95,9 @@ def genImagesOnlyTree(dbCur, nodesWithImgOrPicked, rootName): print("Getting ancestors") nodeMap = genNodeMap(dbCur, nodesWithImgOrPicked, 1e4) print(f"Result has {len(nodeMap)} nodes") + print("Removing composite nodes") + removeCompositeNodes(nodeMap) + print(f"Result has {len(nodeMap)} nodes") print("Removing 'collapsible' nodes") removeCollapsibleNodes(nodeMap, nodesWithImgOrPicked) print(f"Result has {len(nodeMap)} nodes") |