From 6952ec507a6d0909d8ca1e49b1249c78b9c48cd4 Mon Sep 17 00:00:00 2001 From: Alfredo Date: Thu, 16 Nov 2023 16:25:59 -0500 Subject: [PATCH] replace dictionary used to count node input links with collections.Counter --- api/arrange.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/arrange.py b/api/arrange.py index 538e0eb..a71d336 100644 --- a/api/arrange.py +++ b/api/arrange.py @@ -1,6 +1,6 @@ import bpy import typing -from collections import deque +from collections import deque, Counter def _arrange(node_tree, padding: typing.Tuple[float, float] = (50, 25)): # Organize the nodes into columns based on their links. @@ -23,7 +23,7 @@ def _arrange(node_tree, padding: typing.Tuple[float, float] = (50, 25)): return topo_order graph = { node:set() for node in node_tree.nodes } - node_input_link_count = { i:0 for node in node_tree.nodes for i in node.inputs } + node_input_link_count = Counter() for link in node_tree.links: graph[link.from_node].add(link.to_node) node_input_link_count[link.to_socket] += 1