From e25deb5fd30981b6558a09efc47a6defeb9cf0a1 Mon Sep 17 00:00:00 2001 From: Terry Truong Date: Wed, 15 Jun 2022 21:22:44 +1000 Subject: In the reduced-tree, prefer additional-children with images --- backend/data/genReducedTreeData.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/data/genReducedTreeData.py b/backend/data/genReducedTreeData.py index 7dc3acc..ba223b0 100755 --- a/backend/data/genReducedTreeData.py +++ b/backend/data/genReducedTreeData.py @@ -119,7 +119,7 @@ for (name, nodeObj) in nodeMap.items(): children = [row[0] for row in dbCur.execute("SELECT child FROM edges where node = ?", (name,))] newChildren = [n for n in children if not (n in nodeMap or n in namesToRemove) and - compNameRegex.fullmatch(n) == None] + dbCur.execute("SELECT name from node_imgs WHERE name = ?", (n,)).fetchone() != None] newChildNames = newChildren[:max(0, PREF_NUM_CHILDREN - numChildren)] nodeObj["children"].extend(newChildNames) namesToAdd.extend(newChildNames) -- cgit v1.2.3