diff options
Diffstat (limited to 'backend/data/genReducedTreeData.py')
| -rwxr-xr-x | backend/data/genReducedTreeData.py | 2 |
1 files changed, 1 insertions, 1 deletions
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) |